Skip to content

Commit

Permalink
[res] Add TFLiteRecipe for dynamic shape Reshape (#14162)
Browse files Browse the repository at this point in the history
This commit adds TFLiteRecipe for dynamic shape Reshape.

ONE-DCO-1.0-Signed-off-by: Jongwon Yang <[email protected]>
  • Loading branch information
jongwonyang authored Oct 7, 2024
1 parent 49d45bf commit daf9a85
Show file tree
Hide file tree
Showing 2 changed files with 28 additions and 0 deletions.
23 changes: 23 additions & 0 deletions res/TensorFlowLiteRecipes/Inf_Reshape_000/test.recipe
Original file line number Diff line number Diff line change
@@ -0,0 +1,23 @@
operand {
name: "ifm"
type: FLOAT32
shape { dim: 2 dim: 3 dim: 4 }
shape_signature { dim: 2 dim: 3 dim: 4 }
}
operand {
name: "ofm"
type: FLOAT32
shape { dim: 0 dim: 4 }
shape_signature { dim: -1 dim: 4 }
}
operation {
type: "Reshape"
reshape_options {
new_shape: -1
new_shape: 4
}
input: "ifm"
output: "ofm"
}
input: "ifm"
output: "ofm"
5 changes: 5 additions & 0 deletions res/TensorFlowLiteRecipes/Inf_Reshape_000/test.rule
Original file line number Diff line number Diff line change
@@ -0,0 +1,5 @@
# To check if dynamic dimension properly inferred if option has dynamic shape

RULE "VERIFY_FILE_FORMAT" $(verify_file_format) '=' 1

RULE "RESHAPE_SHAPE" $(tensor_shape ofm) '=' [6,4]

0 comments on commit daf9a85

Please sign in to comment.