Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize.cc109 quant_specs_.inference_type = in PrepareQuantizePass()
111 quant_specs_.post_training_quantization = post_training_quantize; in PrepareQuantizePass()
112 quant_specs_.legacy_float_scale = legacy_float_scale; in PrepareQuantizePass()
117 : quant_specs_(quant_specs) {} in PrepareQuantizePass()
137 if (func.getName() == quant_specs_.target_func) { in IsLegalQuantSpecs()
138 return func.getNumArguments() == quant_specs_.input_ranges.size(); in IsLegalQuantSpecs()
148 if (func_name == quant_specs_.target_func) { in GetMinMaxValuesForArgument()
149 return quant_specs_.input_ranges[index]; in GetMinMaxValuesForArgument()
163 QuantizationSpecs quant_specs_; member in mlir::TFL::__anonbf0076700111::PrepareQuantizePass
168 auto& target_func = quant_specs_.target_func; in SetInputNodesQuantizationParams()
[all …]