Searched refs:input_quantization_axis_ (Results 1 – 1 of 1) sorted by relevance
167 &input_quantization_axis_)); in UniformRequantizeOp()171 context, (input_quantization_axis_ >= -1), in UniformRequantizeOp()173 input_quantization_axis_)); in UniformRequantizeOp()180 (!(input_quantization_axis_ >= 0 && output_quantization_axis_ >= 0) || in UniformRequantizeOp()181 input_quantization_axis_ == output_quantization_axis_), in UniformRequantizeOp()196 input_zero_points.shape(), input_quantization_axis_))); in Compute()216 output_zero_points, input_quantization_axis_, in Compute()222 int input_quantization_axis_; member in tensorflow::UniformRequantizeOp