Skip to content

Commit

Permalink
[res] Introduce TransposeConv_002 (#12440)
Browse files Browse the repository at this point in the history
This commit introduces TransposeConv_002 recipe for checking if non-constant out_shape can be inferred by the shape inference.

ONE-DCO-1.0-Signed-off-by: seongwoo <[email protected]>
  • Loading branch information
mhs4670go authored Jan 10, 2024
1 parent 8b8b396 commit 9b2a65e
Showing 1 changed file with 48 additions and 0 deletions.
48 changes: 48 additions & 0 deletions res/TensorFlowLiteRecipes/TransposeConv_002/test.recipe
Original file line number Diff line number Diff line change
@@ -0,0 +1,48 @@
operand {
name: "ifm"
type: FLOAT32
shape { dim: 1 dim: 4 dim: 4 dim: 3 }
}
operand {
name: "out_shape"
type: INT32
shape { dim: 4 }
}
operand {
name: "ker"
type: FLOAT32
shape { dim: 3 dim: 1 dim: 1 dim: 3 }
filler {
tag: "gaussian"
arg: "0.0"
arg: "1.0"
}
}
operand {
name: "ofm"
type: FLOAT32
shape { dim: 1 dim: 4 dim: 4 dim: 3 }
}
operation {
type: "Shape"
shape_options {
out_type: INT32
}
input: "ifm"
output: "out_shape"
}
operation {
type: "TransposeConv"
transpose_conv_options {
padding: SAME
stride_w: 1
stride_h: 1
activation: NONE
}
input: "out_shape"
input: "ker"
input: "ifm"
output: "ofm"
}
input: "ifm"
output: "ofm"

0 comments on commit 9b2a65e

Please sign in to comment.