diff options
Diffstat (limited to 'compiler/tflchef/tests/explicit_datachef/test.recipe')
-rw-r--r-- | compiler/tflchef/tests/explicit_datachef/test.recipe | 28 |
1 files changed, 28 insertions, 0 deletions
diff --git a/compiler/tflchef/tests/explicit_datachef/test.recipe b/compiler/tflchef/tests/explicit_datachef/test.recipe new file mode 100644 index 000000000..bd5213f39 --- /dev/null +++ b/compiler/tflchef/tests/explicit_datachef/test.recipe @@ -0,0 +1,28 @@ +operand { + name: "ifm" + type: FLOAT32 + shape { dim: 1 dim: 1 dim: 1 dim: 10 } +} +operand { + name: "shape" + type: INT32 + shape { dim: 2 } + filler { tag: "explicit" arg: "-1" arg: "10" } +} +operand { + name: "ofm" + type: FLOAT32 + shape { dim: 1 dim: 10 } +} +operation { + type: "Reshape" + reshape_options { + new_shape: -1 + new_shape: 10 + } + input: "ifm" + input: "shape" + output: "ofm" +} +input: "ifm" +output: "ofm" |