Searched defs:inference_type (Results 1 – 12 of 12) sorted by relevance
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
D | quantization_config.cc | 45 absl::string_view inference_type, in ParseInputNodeQuantSpecs() 86 tensorflow::DataType inference_type, QuantizationSpecs* quant_specs) { in GetInputNodeQuantSpecs()
|
D | quantization_config.h | 70 tensorflow::DataType inference_type = tensorflow::DT_FLOAT; member
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
D | tf_mlir_translate_cl.cc | 57 opt<std::string> inference_type( variable
|
/external/tensorflow/tensorflow/lite/python/ |
D | wrap_toco.py | 47 fully_quantize, inference_type, argument
|
D | convert.py | 129 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/ |
D | toco_python_api_wrapper.cc | 60 bool fully_quantize, int inference_type, bool enable_numeric_verify) { in PYBIND11_MODULE()
|
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
D | quantize_model.cc | 43 const tflite::TensorType& inference_type, in QuantizeModel()
|
/external/tensorflow/tensorflow/lite/toco/ |
D | toco_flags.proto | 90 optional IODataType inference_type = 4; field
|
D | toco_tooling.cc | 247 const IODataType inference_type = toco_flags.inference_type(); in TransformWithStatus() local
|
D | args.h | 170 Arg<std::string> inference_type; member
|
/external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
D | tf_tfl_flatbuffer_helpers.cc | 203 tensorflow::DataType inference_type = in PopulateQuantizationSpecs() local
|
/external/tensorflow/tensorflow/lite/toco/python/ |
D | toco_python_api.cc | 239 bool fully_quantize, int inference_type, in MlirQuantizeModel()
|