Home
last modified time | relevance | path

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

/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()
137 const int quantization_axis = input_quantization_axis >= 0 in EvalRequantize()
138 ? input_quantization_axis in EvalRequantize()
/aosp_15_r20/external/tensorflow/tensorflow/core/api_def/base_api/
H A Dapi_def_UniformRequantize.pbtxt57 name: "input_quantization_axis"
102 …ut`, which was quantized using {input_scales, input_zero_points, input_quantization_axis, input_qu…
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/ops/compat/ops_history_v2/
H A DUniformRequantize.pbtxt48 name: "input_quantization_axis"
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/
H A Dtf_quant_ops.td56 DefaultValuedAttr<I64Attr, "-1">:$input_quantization_axis,
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/
H A Dtensorflow.raw_ops.pbtxt5309 …utput_quantization_min_val\', \'output_quantization_max_val\', \'input_quantization_axis\', \'outp…
/aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/
H A Dtensorflow.raw_ops.pbtxt5309 …utput_quantization_min_val\', \'output_quantization_max_val\', \'input_quantization_axis\', \'outp…
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dops.pbtxt61625 name: "input_quantization_axis"