Searched refs:op_quant_scale_spec_getter (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | quantization_utils.cc | 744 OpQuantScaleSpecGetter op_quant_scale_spec_getter) { in IsStatsRedundant() argument
747 op_quant_scale_spec_getter(op)->has_fixed_output_range; in IsStatsRedundant()
751 OpQuantScaleSpecGetter op_quant_scale_spec_getter) { in IsSameScaleOp() argument
754 op_quant_scale_spec_getter(op)->has_same_scale_requirement; in IsSameScaleOp()
759 OpQuantScaleSpecGetter op_quant_scale_spec_getter) { in RemoveRedundantStatsOps() argument
790 op_quant_scale_spec_getter)) { in RemoveRedundantStatsOps()
798 if (!IsSameScaleOp(user, op_quant_scale_spec_getter) || in RemoveRedundantStatsOps()
830 if (!IsSameScaleOp(def, op_quant_scale_spec_getter)) { in RemoveRedundantStatsOps()
|
| H A D | quantization_driver.cc | 108 OpQuantScaleSpecGetter op_quant_scale_spec_getter, in QuantizationDriver() argument
115 op_quant_scale_spec_getter_(op_quant_scale_spec_getter), in QuantizationDriver()
1183 OpQuantScaleSpecGetter op_quant_scale_spec_getter, bool infer_tensor_ranges, in ApplyQuantizationParamsPropagation() argument
1186 op_quant_scale_spec_getter, infer_tensor_ranges, in ApplyQuantizationParamsPropagation()
|
| H A D | quantization_utils.h | 881 OpQuantScaleSpecGetter op_quant_scale_spec_getter, bool infer_tensor_ranges,
894 OpQuantScaleSpecGetter op_quant_scale_spec_getter =
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/passes/ |
| H A D | quantize.cc | 154 MLIRContext* context, OpQuantScaleSpecGetter op_quant_scale_spec_getter) in QuantizeSameScaleOpsPattern() argument
158 op_quant_scale_spec_getter_(op_quant_scale_spec_getter) {} in QuantizeSameScaleOpsPattern()
|