-
Notifications
You must be signed in to change notification settings - Fork 157
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[res] Add TFLiteRecipe for dynamic shape pad #14109
Conversation
This commit adds TFLiteRecipe for dynamic shape pad. ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]
change padding data type to int32 Co-authored-by: SaeHie Park <[email protected]>
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
operand { | ||
name: "ifm" | ||
type: FLOAT32 | ||
shape { dim: 1 dim: 1 dim: 3 dim: 2 } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just curious,
Is there any special reason to set the second dim
to 1
?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just curious, Is there any special reason to set the second
dim
to1
?
There is no particular reason. I just wanted to check if rule-lib.sh
works well in dim
s other than the first one.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Just curious, Is there any special reason to set the second
dim
to1
?
Oh, I misunderstand your question. 😂
I referred this code for shape_signature
to test second dim
to dynamic.
ONE/compiler/tflchef/tests/shape_signature/test.recipe
Lines 4 to 5 in 06a536b
shape { dim: 1 dim: 3 dim: 3 dim: 2 } | |
shape_signature { dim: -1 dim: 3 dim: 3 dim: 2 } |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM!
=)
This commit adds TFLiteRecipe for dynamic shape pad.
ONE-DCO-1.0-Signed-off-by: JuYoung Lee [email protected]