diff options
Diffstat (limited to 'res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000')
-rw-r--r-- | res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.recipe | 29 | ||||
-rw-r--r-- | res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.rule | 6 |
2 files changed, 35 insertions, 0 deletions
diff --git a/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.recipe b/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.recipe new file mode 100644 index 000000000..b84058b0e --- /dev/null +++ b/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.recipe @@ -0,0 +1,29 @@ +operand { + name: "ifm" + type: FLOAT32 + shape { dim: 1 dim: 16 dim: 1 dim: 1 } +} +operand { + name: "t1" + type: FLOAT32 + shape { dim: 1 dim: 16 dim: 1 } +} +operand { + name: "ofm" + type: FLOAT32 + shape { dim: 1 dim: 16 } +} +operation { + type: "Squeeze" + squeeze_options { squeeze_dim: 3 } + input: "ifm" + output: "t1" +} +operation { + type: "Squeeze" + squeeze_options { squeeze_dim: 2 } + input: "t1" + output: "ofm" +} +input: "ifm" +output: "ofm" diff --git a/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.rule b/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.rule new file mode 100644 index 000000000..66a105a73 --- /dev/null +++ b/res/TensorFlowLiteRecipes/Net_Squeeze_Squeeze_000/test.rule @@ -0,0 +1,6 @@ +# To check if Squeeze is substituted to Reshape op + +RULE "VERIFY_FILE_FORMAT" $(verify_file_format) '=' 1 + +RULE "SQUEEZE_COUNT" $(op_count SQUEEZE) '=' 0 +RULE "RESHAPE_COUNT" $(op_count RESHAPE) '=' 2 |