-
Notifications
You must be signed in to change notification settings - Fork 3.5k
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
[TFLite, QNN] Slice op #6217
[TFLite, QNN] Slice op #6217
Conversation
@siju-samuel Please review. |
assert len(output_tensors) == 1, "There should be only 1 output tensor" | ||
output_tensor = output_tensors[0] | ||
assert self.has_same_qnn_params(input_tensor, output_tensor), \ | ||
"TFLite reshape requires input and output scale and zero points to be equal" |
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.
Tflite reshape -> Tflite slice
input_range = {'in_data': (-100, 100)} if quantized else None | ||
compare_tflite_with_tvm([data], ['in_data:0'], [in_data], [out], |
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.
This 2 common statements, put outside if-else block.
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.
@anijain2305 please followup the review comments.
LGTM modulo @siju-samuel 's comments. |
TFLite quantized slice op has same input and output qnn params. Just adding a check and a test case.
@d-smirnov Please take a look at this PR if it can help simplify - #6018 My goal was to keep the changes minimal here.
@u99127