Home
last modified time | relevance | path

Searched refs:max_quantized (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dquantize_and_dequantize_op.cc121 xla::XlaOp min_quantized, max_quantized; in Compile() local
133 max_quantized = in Compile()
139 max_quantized = Pow(two, ConvertElementType(num_bits, xla_type)) - one; in Compile()
149 Select(Gt(max_quantized * max_range, zero), max_quantized / max_range, in Compile()
156 Select(cond, min_range / min_quantized, max_range / max_quantized); in Compile()
158 max_range = Select(cond, max_quantized * inverse_scale, max_range); in Compile()
/external/tensorflow/tensorflow/core/kernels/
Dquantize_and_dequantize_op.h155 const int64 max_quantized = in ComputeQuantizationRange() local
163 const T scale_from_max_side = (max_quantized * *max_range > 0) in ComputeQuantizationRange()
164 ? max_quantized / *max_range in ComputeQuantizationRange()
171 *max_range = max_quantized * *inverse_scale; in ComputeQuantizationRange()
174 *inverse_scale = *max_range / max_quantized; in ComputeQuantizationRange()