blob: 2dd24af6096873f502f9466c2c5f4974f848299d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
|
## EXAMPLE
#
# Add(RECIPE_REPO PASS pass1 pass2 ...)
#
## SUPPORTED PASS
#
# fuse_instnorm
# resolve_customop_batchmatmul
# resolve_customop_matmul
## TFLITE RECIPE
Add(Net_Preactivation_BN_000 PASS fuse_preactivation_batchnorm)
Add(Net_BroadcastTo_AddV2_000 PASS resolve_customop_add)
Add(Net_BroadcastTo_AddV2_001 PASS resolve_customop_add)
Add(Net_Conv_Add_Mul_000 PASS fuse_batchnorm_with_conv)
Add(Net_Conv_Add_Mul_001 PASS fuse_batchnorm_with_conv)
Add(Net_Conv_Add_Mul_002 PASS fuse_batchnorm_with_conv)
Add(Net_Conv_FakeQuant_000 PASS remove_fakequant)
Add(Net_Conv_QuantDequant_000 PASS remove_quantdequant)
Add(Net_Conv_Min_Max_000 PASS transform_min_max_to_relu6)
Add(Net_Conv_Min_Relu_000 PASS transform_min_relu_to_relu6)
Add(Net_Conv_PReluGraph_000 PASS fuse_prelu)
Add(Net_Conv_Relu6_000 PASS fuse_activation_function)
Add(Net_Duplicate_Weights_000 PASS remove_duplicate_const)
Add(Net_DwConv_BN_000 PASS fuse_batchnorm_with_dwconv)
Add(Net_DwConv_BN_001 PASS fuse_batchnorm_with_dwconv)
Add(Net_FullyConnected_Add_000 PASS fold_fully_connected)
Add(Net_Reshape_Reshape_000 PASS remove_redundant_reshape)
Add(Net_Squeeze_Squeeze_000 PASS substitute_squeeze_to_reshape)
Add(Net_TConv_Add_000 PASS fuse_add_with_tconv)
Add(Net_TConv_Add_001 PASS fuse_add_with_tconv)
Add(Net_TConv_Add_002 PASS fuse_add_with_tconv)
Add(Net_TConv_BN_000 PASS fuse_batchnorm_with_tconv)
Add(Net_TConv_BN_001 PASS fuse_batchnorm_with_tconv)
Add(Net_TConv_BN_002 PASS fuse_batchnorm_with_tconv)
Add(Net_TConv_BN_003 PASS fuse_batchnorm_with_tconv)
Add(Net_TConv_BN_004 PASS fuse_batchnorm_with_tconv)
Add(Net_TConv_BN_005 PASS fuse_batchnorm_with_tconv)
Add(Net_InstanceNorm_001 PASS fuse_instnorm)
Add(Net_InstanceNorm_003 PASS fuse_instnorm)
Add(Net_InstanceNorm_004 PASS fuse_instnorm)
Add(Net_InstanceNorm_005 PASS fuse_instnorm)
Add(Net_InstanceNorm_006 PASS fuse_instnorm)
Add(Net_InstanceNorm_007 PASS fuse_instnorm)
Add(Net_Maximum_Minimum_000 PASS transform_min_max_to_relu6)
Add(BatchMatMulV2_000 PASS resolve_customop_batchmatmul)
Add(MatMul_000 PASS resolve_customop_matmul)
Add(DepthwiseConv2D_003 PASS)
Add(PadV2_001 PASS substitute_padv2_to_pad)
Add(StridedSlice_003 PASS substitute_strided_slice_to_reshape)
Add(MaxPoolWithArgmax_000 PASS resolve_customop_max_pool_with_argmax)
Add(MaxPoolWithArgmax_001 PASS resolve_customop_max_pool_with_argmax)
Add(MaxPoolWithArgmax_002 PASS resolve_customop_max_pool_with_argmax)
Add(FullyConnected_007 PASS replace_non_const_fc_with_batch_matmul)
Add(FullyConnected_008 PASS replace_non_const_fc_with_batch_matmul)
Add(Net_Gelu_000 PASS fuse_gelu)
Add(Net_Gelu_001 PASS fuse_gelu)
Add(HardSwish_001 PASS decompose_hardswish)
## CIRCLE RECIPE
Add(CircleBatchMatMul_000)
# REGRESSION test
Add(REGRESS_ONNX_Conv_BN_001 PASS
convert_nchw_to_nhwc
nchw_to_nhwc_input_shape
nchw_to_nhwc_output_shape
remove_redundant_transpose
substitute_transpose_to_reshape
remove_redundant_reshape
remove_unnecessary_reshape
fuse_batchnorm_with_conv)
Add(REGRESS_ONNX_Conv_BN_Relu6_001 PASS
convert_nchw_to_nhwc
nchw_to_nhwc_input_shape
nchw_to_nhwc_output_shape
remove_redundant_transpose
transform_min_max_to_relu6
fuse_batchnorm_with_conv
fuse_activation_function)
Add(REGRESS_ONNX_Conv_BN_MeanMean_001 PASS
convert_nchw_to_nhwc
nchw_to_nhwc_input_shape
nchw_to_nhwc_output_shape
remove_redundant_transpose
fuse_batchnorm_with_conv
fuse_activation_function
fuse_mean_with_mean
fuse_transpose_with_mean)
Add(REGRESS_ONNX_Mul_Mul_000 PASS
convert_nchw_to_nhwc)
|
