Home
last modified time | relevance | path

Searched defs:inference_type (Results 1 – 12 of 12) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_config.cc45 absl::string_view inference_type, in ParseInputNodeQuantSpecs()
86 tensorflow::DataType inference_type, QuantizationSpecs* quant_specs) { in GetInputNodeQuantSpecs()
Dquantization_config.h70 tensorflow::DataType inference_type = tensorflow::DT_FLOAT; member
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dtf_mlir_translate_cl.cc57 opt<std::string> inference_type( variable
/external/tensorflow/tensorflow/lite/python/
Dwrap_toco.py47 fully_quantize, inference_type, argument
Dconvert.py129 inference_type=_types_pb2.INT8, argument
309 def build_toco_flags(inference_type=dtypes.float32, argument
365 inference_type=dtypes.float32, argument
/external/tensorflow/tensorflow/python/lite/
Dtoco_python_api_wrapper.cc60 bool fully_quantize, int inference_type, bool enable_numeric_verify) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dquantize_model.cc43 const tflite::TensorType& inference_type, in QuantizeModel()
/external/tensorflow/tensorflow/lite/toco/
Dtoco_flags.proto90 optional IODataType inference_type = 4; field
Dtoco_tooling.cc247 const IODataType inference_type = toco_flags.inference_type(); in TransformWithStatus() local
Dargs.h170 Arg<std::string> inference_type; member
/external/tensorflow/tensorflow/compiler/mlir/lite/python/
Dtf_tfl_flatbuffer_helpers.cc203 tensorflow::DataType inference_type = in PopulateQuantizationSpecs() local
/external/tensorflow/tensorflow/lite/toco/python/
Dtoco_python_api.cc239 bool fully_quantize, int inference_type, in MlirQuantizeModel()