Permalink
Show file tree
Hide file tree
66 changes: 66 additions & 0 deletions
66
tensorflow/python/kernel_tests/quantization_ops/quantization_ops_test.py
1 comment
on commit
sign in to comment.
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
Browse files
Browse the repository at this point in the history
Fix QuantizedAvgPool invalid rank issue.
Passing in an invalid rank in eager mode led to a runtime crash. Updated the runtime error to match the one generated by the shape function for better consistency. PiperOrigin-RevId: 462419362
- Loading branch information
1 parent
177036a
commit 7cdf9d4
Showing
3 changed files
with
100 additions
and
22 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
7cdf9d4There 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 is causing a unit test failure on --config=mkl_aarch64
#56861