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.
|