Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dtf_quant_ops.td59 DefaultValuedAttr<I64Attr, "-1">:$output_quantization_axis,
92 DefaultValuedAttr<I64Attr, "-1">:$output_quantization_axis,
155 DefaultValuedAttr<I64Attr, "-1">:$output_quantization_axis,
190 DefaultValuedAttr<I64Attr, "-1">:$output_quantization_axis,
226 DefaultValuedAttr<I64Attr, "-1">:$output_quantization_axis,
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_UniformRequantize.pbtxt83 name: "output_quantization_axis"
103 …r, which is quantized using {output_scales, output_zero_points, output_quantization_axis, output_q…
113 * per-axis -> per-axis where input_quantization_axis equals output_quantization_axis.
114 i.e. At least one among input_quantization_axis and output_quantization_axis must be -1, or two mus…
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/uniform_quant_ops/
H A Duniform_requantize_op.cc127 int input_quantization_axis, int output_quantization_axis, in EvalRequantize() argument
130 if (input_quantization_axis == -1 && output_quantization_axis == -1) { in EvalRequantize()
139 : output_quantization_axis; in EvalRequantize()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/
H A DUniformRequantize.pbtxt63 name: "output_quantization_axis"
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt5309 …\'output_quantization_max_val\', \'input_quantization_axis\', \'output_quantization_axis\', \'name…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt5309 …\'output_quantization_max_val\', \'input_quantization_axis\', \'output_quantization_axis\', \'name…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt61640 name: "output_quantization_axis"