diff options
Diffstat (limited to 'compiler/tflchef/tests/make_sparse_f16/test.recipe')
-rw-r--r-- | compiler/tflchef/tests/make_sparse_f16/test.recipe | 54 |
1 files changed, 54 insertions, 0 deletions
diff --git a/compiler/tflchef/tests/make_sparse_f16/test.recipe b/compiler/tflchef/tests/make_sparse_f16/test.recipe new file mode 100644 index 000000000..5977a1d32 --- /dev/null +++ b/compiler/tflchef/tests/make_sparse_f16/test.recipe @@ -0,0 +1,54 @@ +operand { + name: "in" + type: FLOAT32 + shape { dim: 4 dim: 4 } +} +operand { + name: "sparse16" + type: FLOAT16 + shape { dim: 4 dim: 4 } + filler { + tag: "explicit" + arg: "2" arg: "0" arg: "0" arg: "0" + arg: "0" arg: "0" arg: "0" arg: "0" + arg: "0" arg: "0" arg: "0" arg: "0" + arg: "0" arg: "0" arg: "0" arg: "3" + } + make_sparse: true +} +operand { + name: "dense16" + type: FLOAT16 + shape { dim: 4 dim: 4 } +} +operand { + name: "dense32" + type: FLOAT32 + shape { dim: 4 dim: 4 } +} +operand { + name: "out" + type: FLOAT32 + shape { dim: 4 dim: 4 } +} +operation { + type: "Densify" + input: "sparse16" + output: "dense16" +} +operation { + type: "Dequantize" + input: "dense16" + output: "dense32" +} +operation { + type: "Add" + input: "in" + input: "dense32" + output: "out" + add_options { + activation: NONE + } +} +input: "in" +output: "out" |