-
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 StridedSlice #14156
[res] Add TFLiteRecipe for dynamic shape StridedSlice #14156
Conversation
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.
I left a comment.
Anyway the code LGTM!
=)
name: "ofm" | ||
type: FLOAT32 | ||
shape { dim: 1 dim: 0 dim: 0 dim: 5 } | ||
shape_signature { dim: 1 dim: -1 dim: -1 dim: 5 } |
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.
I'll leave this comment for the record.
Theoretically, in this case, it seems that the shape of the output tenser
can be determined by the begin
, end
, and stride
values.
I think this can be solved by updating the tflite
code part together later.
I will create a separate issue related to this.
This commit adds TFLiteRecipe for dynamic shape StridedSlice. ONE-DCO-1.0-Signed-off-by: sunki <[email protected]>
f38b858
to
6b1d03e
Compare
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.
👍
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 StridedSlice.
ONE-DCO-1.0-Signed-off-by: sunki [email protected]