tf.quantize_and_dequantize_v4
Returns the gradient of QuantizeAndDequantizeV4
.
tf.quantize_and_dequantize_v4( input, input_min, input_max, signed_input=True, num_bits=8, range_given=False, round_mode='HALF_TO_EVEN', narrow_range=False, axis=-1, name=None )
This is almost identical to QuantizeAndDequantizeV2, except that it returns a gradient of 1 for inputs that are within the quantization range, or 0 otherwise.
Args | |
---|---|
input | A Tensor . Must be one of the following types: bfloat16 , half , float32 , float64 . |
input_min | A Tensor . Must have the same type as input . |
input_max | A Tensor . Must have the same type as input . |
signed_input | An optional bool . Defaults to True . |
num_bits | An optional int . Defaults to 8 . |
range_given | An optional bool . Defaults to False . |
round_mode | An optional string from: "HALF_TO_EVEN", "HALF_UP" . Defaults to "HALF_TO_EVEN" . |
narrow_range | An optional bool . Defaults to False . |
axis | An optional int . Defaults to -1 . |
name | A name for the operation (optional). |
Returns | |
---|---|
A Tensor . Has the same type as input . |
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/quantize_and_dequantize_v4