| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/common/ |
| D | targets.h | 61 inline std::string GetInferenceString(InferenceType inference_type) { in GetInferenceString()
96 auto inference_type = op->getAttrOfType<StringAttr>(kInferenceType); in GetInferenceTypeAnnotation() local
106 InferenceType inference_type; member
132 auto inference_type = GetInferenceTypeAnnotation(op); in GetInferenceDeviceTypeForOp() local
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/hardwares/ |
| D | gpu_hardware.cc | 86 InferenceType inference_type = GetInferenceType(op); in IsOpSupported() local
108 InferenceType inference_type = GetInferenceType(op); in IsOpSupported() local
130 InferenceType inference_type = GetInferenceType(op); in IsOpSupported() local
154 InferenceType inference_type = GetInferenceType(op); in IsOpSupported() local
|
| D | cpu_hardware.cc | 40 inline float InferenceTypeEfficiency(InferenceType inference_type) { in InferenceTypeEfficiency()
|
| /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 | 73 tensorflow::DataType inference_type = tensorflow::DT_FLOAT; member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | tf_mlir_translate_cl.cc | 58 opt<std::string> inference_type( variable
|
| /external/tensorflow/tensorflow/lite/python/ |
| D | convert.py | 204 inference_type=_types_pb2.QUANTIZED_INT8, argument
429 def build_toco_flags(inference_type=dtypes.float32, argument
498 inference_type=dtypes.float32, argument
|
| D | wrap_toco.py | 41 fully_quantize, inference_type, argument
|
| D | lite.py | 634 inference_type=None, argument
|
| /external/tensorflow/tensorflow/python/lite/ |
| D | toco_python_api_wrapper.cc | 58 py::object node_blocklist) { in PYBIND11_MODULE()
|
| /external/tensorflow/tensorflow/lite/toco/python/ |
| D | toco_python_api.cc | 228 tflite::TensorType FromTocoDataTypeToTflitToTensorType(int inference_type) { in FromTocoDataTypeToTflitToTensorType()
278 bool fully_quantize, int inference_type, in MlirQuantizeModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/ |
| D | quantize_model.cc | 51 const tflite::TensorType& inference_type, in QuantizeModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | target_annotation.cc | 80 const InferenceType inference_type = GetInferenceType(op); in SetTargetAnnotation() local
|
| D | get_alternative_subgraph.cc | 65 InferenceType inference_type, ArrayRef<std::string> devices) { in GetAllAlternativeInferenceDeviceType()
|
| /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 | 210 tensorflow::DataType inference_type = in PopulateQuantizationSpecs() local
|