Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_requantize_op.cc62 const bool input_per_axis_quantization = input_scales.dims() == 1; in PerAxisRequantize() local
64 const auto& per_axis_scales_shape = input_per_axis_quantization in PerAxisRequantize()
86 input_scales_data[input_per_axis_quantization ? i : 0]) / in PerAxisRequantize()
108 input_per_axis_quantization, output_per_axis_quantization, in PerAxisRequantize()
113 input_zero_points_data[input_per_axis_quantization ? i : 0], in PerAxisRequantize()