Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_config.cc73 tensorflow::DataType final_type = tensorflow::DT_FLOAT; in ParseInputNodeQuantSpecs() local
75 !DataType_Parse(std::string(inference_type), &final_type)) { in ParseInputNodeQuantSpecs()
78 return GetInputNodeQuantSpecs(input_nodes, node_mins, node_maxs, final_type, in ParseInputNodeQuantSpecs()
Dquantization_utils.cc181 Type final_type = GetQuantizedType( in GetQuantizedTypeAttr() local
184 if (!final_type) return {}; in GetQuantizedTypeAttr()
185 return TypeAttr::get(final_type); in GetQuantizedTypeAttr()
266 Type final_type = qtype.castFromExpressedType(target); in CastQuantizedTypeAttrFromExpressedType() local
267 if (!final_type) return {}; in CastQuantizedTypeAttrFromExpressedType()
268 return TypeAttr::get(final_type); in CastQuantizedTypeAttrFromExpressedType()
Dquantization_driver.cc431 Type final_type; in SetConstantResultParams() local
444 final_type = GetUniformQuantizedPerAxisTypeForWeight( in SetConstantResultParams()
449 final_type = GetUniformQuantizedTypeForWeight( in SetConstantResultParams()
454 if (auto quant_type = final_type.dyn_cast_or_null<quant::QuantizedType>()) { in SetConstantResultParams()