Skip to content

Commit

Permalink
[res] Introduce Reshape_004 tflite recipe (#14151)
Browse files Browse the repository at this point in the history
This will introduce Reshape_004 tflite recipe with 0 in shape.

ONE-DCO-1.0-Signed-off-by: SaeHie Park <[email protected]>
  • Loading branch information
seanshpark authored Oct 2, 2024
1 parent beebec9 commit e07380e
Showing 1 changed file with 31 additions and 0 deletions.
31 changes: 31 additions & 0 deletions res/TensorFlowLiteRecipes/Reshape_004/test.recipe
Original file line number Diff line number Diff line change
@@ -0,0 +1,31 @@
# Test model of Reshape with 0 in shape.
# May not work in interpreter.
operand {
name: "ifm"
type: FLOAT32
shape { dim: 1 dim: 3 dim: 2 dim: 3 }
}
operand {
name: "shape"
type: INT32
shape { dim: 3 }
filler { tag: "explicit" arg: "0" arg: "3" arg: "-1" }
}
operand {
name: "ofm"
type: FLOAT32
shape { dim: 1 dim: 3 dim: 6 }
}
operation {
type: "Reshape"
reshape_options {
new_shape: 0
new_shape: 3
new_shape: -1
}
input: "ifm"
input: "shape"
output: "ofm"
}
input: "ifm"
output: "ofm"

0 comments on commit e07380e

Please sign in to comment.