Skip to content
Permalink
Browse files Browse the repository at this point in the history
Validate the rank and number of elements of the num_bits tensor for…
… QuantizeAndDequantizeV3.

QuantizeAndDequantizeV3Op, which accepts `num_bits` as a tensor, has a precondition that it should be rank <= 1 and the number of elements should be 1.
This change adds a validation for the Compute() method for this condition.

PiperOrigin-RevId: 463755293
  • Loading branch information
dansuh17 authored and tensorflower-gardener committed Jul 28, 2022
1 parent a49a3e4 commit f3f9cb3
Show file tree
Hide file tree
Showing 2 changed files with 162 additions and 105 deletions.

0 comments on commit f3f9cb3

Please sign in to comment.