Home
last modified time | relevance | path

Searched defs:quantization_axis (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_dequantize_op.cc33 const Tensor& zero_points, int quantization_axis, in EvalPerChannelDequantize()
52 const Tensor& zero_points, int quantization_axis, in EvalDequantize()
H A Duniform_quantize_op.cc35 const Tensor& zero_points, int quantization_axis, in EvalPerChannelQuantize()
57 const Tensor& zero_points, int quantization_axis, in EvalQuantize()
H A Duniform_requantize_op.cc60 int quantization_axis, int32_t quantization_min_val, in PerAxisRequantize()
137 const int quantization_axis = input_quantization_axis >= 0 in EvalRequantize() local
H A Dtensor_utils.cc24 int quantization_axis) { in QuantizationAxisAndShapeValid()