Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_requantize_op.cc169 &output_quantization_axis_)); in UniformRequantizeOp()
175 context, (output_quantization_axis_ >= -1), in UniformRequantizeOp()
177 output_quantization_axis_)); in UniformRequantizeOp()
180 (!(input_quantization_axis_ >= 0 && output_quantization_axis_ >= 0) || in UniformRequantizeOp()
181 input_quantization_axis_ == output_quantization_axis_), in UniformRequantizeOp()
200 output_zero_points.shape(), output_quantization_axis_))); in Compute()
217 output_quantization_axis_, output_quantization_min_val_, in Compute()
225 int output_quantization_axis_; member in tensorflow::UniformRequantizeOp