-
Notifications
You must be signed in to change notification settings - Fork 74k
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
[Intel MKL] Enabling quantized pooling ops with native format #45484
[Intel MKL] Enabling quantized pooling ops with native format #45484
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.
Thank you for the PR and I'm so sorry for the delay! I have minor comments.
Tensor output_quantized; | ||
conv_comp.ConvertMKL2TF<quint8>(DT_QUINT8, output, mkl_shape_tensor, | ||
output_quantized); | ||
if (!NativeFormatEnabled()) { |
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 part can use the refactored function mentioned in #45108.
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 used the refactored function form #45108.
Thank you for the comments. I addressed them. |
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.
Thank you for the changes! I have one minor nit.
NativeFormatEnabled() ? output : output_quantized, output_min, | ||
output_max); | ||
|
||
const Tensor* mkl_shape_tensor = |
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.
Nit: Can we add _ptr
to the name? Same for line 180.
const Tensor* mkl_shape_tensor = | |
const Tensor* mkl_shape_tensor_ptr = |
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.
Sure. Done.
I have addressed the comment. Thank you! |
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.
Thank you again for the PR!
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.
Thank you for the PR and sorry for the delay!
No description provided.