| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/common/ |
| D | targets.h | 63 inline std::string GetInferenceString(InferenceType inference_type) { in GetInferenceString()
98 auto inference_type = op->getAttrOfType<StringAttr>(kInferenceType); in GetInferenceTypeAnnotation() local
108 InferenceType inference_type; member
134 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/lite/ |
| D | quantize_weights.cc | 80 const tflite::TensorType& inference_type, const StringSet& denylisted_ops, in QuantizeWeights()
192 tflite::TensorType inference_type; in QuantizeWeights() local
216 const tflite::TensorType inference_type = tflite::TensorType_INT8; in QuantizeWeights() local
|
| D | quantize_model.cc | 53 const tflite::TensorType& inference_type, in QuantizeModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_config.cc | 93 absl::string_view inference_type, in ParseInputNodeQuantSpecs()
134 tensorflow::DataType inference_type, QuantizationSpecs* quant_specs) { in GetInputNodeQuantSpecs()
|
| D | quantization_config.h | 99 tensorflow::DataType inference_type = tensorflow::DT_FLOAT; member
|
| D | quantization_utils.h | 391 tensorflow::DataType inference_type = in matchAndRewrite() local
|
| /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 | wrap_toco.py | 37 fully_quantize, inference_type, argument
|
| D | convert.py | 202 inference_type=_types_pb2.QUANTIZED_INT8, argument
471 def build_conversion_flags(inference_type=dtypes.float32, argument
|
| D | lite.py | 761 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 | 242 tflite::TensorType FromTocoDataTypeToTflitToTensorType(int inference_type) { in FromTocoDataTypeToTflitToTensorType()
292 bool fully_quantize, int inference_type, in MlirQuantizeModel()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/ |
| D | target_annotation.cc | 85 const InferenceType inference_type = GetInferenceType(op); in SetTargetAnnotation() local
|
| D | get_alternative_subgraph.cc | 68 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 | 248 const IODataType inference_type = toco_flags.inference_type(); in TransformWithStatus() local
|
| D | args.h | 172 Arg<std::string> inference_type; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/python/ |
| D | tf_tfl_flatbuffer_helpers.cc | 216 tensorflow::DataType inference_type = in PopulateQuantizationSpecs() local
|