summaryrefslogtreecommitdiff
path: root/compiler/tflchef/tests/make_sparse/test.recipe
diff options
context:
space:
mode:
Diffstat (limited to 'compiler/tflchef/tests/make_sparse/test.recipe')
-rw-r--r--compiler/tflchef/tests/make_sparse/test.recipe44
1 files changed, 44 insertions, 0 deletions
diff --git a/compiler/tflchef/tests/make_sparse/test.recipe b/compiler/tflchef/tests/make_sparse/test.recipe
new file mode 100644
index 000000000..15cc93a5d
--- /dev/null
+++ b/compiler/tflchef/tests/make_sparse/test.recipe
@@ -0,0 +1,44 @@
+operand {
+ name: "in"
+ type: FLOAT32
+ shape { dim: 4 dim: 4 }
+}
+operand {
+ name: "sparse"
+ type: FLOAT32
+ 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: "dense"
+ type: FLOAT32
+ shape { dim: 4 dim: 4 }
+}
+operand {
+ name: "out"
+ type: FLOAT32
+ shape { dim: 4 dim: 4 }
+}
+operation {
+ type: "Densify"
+ input: "sparse"
+ output: "dense"
+}
+operation {
+ type: "Add"
+ input: "in"
+ input: "dense"
+ output: "out"
+ add_options {
+ activation: NONE
+ }
+}
+input: "in"
+output: "out"