Home
last modified time | relevance | path

Searched refs:TfLiteType (Results 1 – 25 of 42) sorted by relevance

12

/external/tensorflow/tensorflow/lite/
Dinterpreter.h37 constexpr TfLiteType typeToTfLiteType() { in typeToTfLiteType()
41 constexpr TfLiteType typeToTfLiteType<int>() {
45 constexpr TfLiteType typeToTfLiteType<int16_t>() {
49 constexpr TfLiteType typeToTfLiteType<int64_t>() {
53 constexpr TfLiteType typeToTfLiteType<float>() {
57 constexpr TfLiteType typeToTfLiteType<unsigned char>() {
61 constexpr TfLiteType typeToTfLiteType<int8_t>() {
65 constexpr TfLiteType typeToTfLiteType<bool>() {
69 constexpr TfLiteType typeToTfLiteType<std::complex<float>>() {
73 constexpr TfLiteType typeToTfLiteType<string>() {
[all …]
Dinterpreter.cc156 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadOnly()
165 int tensor_index, TfLiteType type, const char* name, in SetTensorParametersReadWrite()
174 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadOnly()
188 int tensor_index, TfLiteType type, const char* name, const size_t rank, in SetTensorParametersReadWrite()
Doptional_debug_tools.cc37 const char* TensorTypeName(TfLiteType type) { in TensorTypeName()
/external/tensorflow/tensorflow/lite/c/
Dbuiltin_op_data.h265 TfLiteType in_data_type;
266 TfLiteType out_data_type;
315 TfLiteType output_type;
319 TfLiteType output_type;
333 TfLiteType out_type;
368 TfLiteType index_out_type;
Dc_api_internal.h207 } TfLiteType; typedef
210 const char* TfLiteTypeGetName(TfLiteType type);
289 TfLiteType type;
350 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims,
Dc_api_internal.c120 void TfLiteTensorReset(TfLiteType type, const char* name, TfLiteIntArray* dims, in TfLiteTensorReset()
153 const char* TfLiteTypeGetName(TfLiteType type) { in TfLiteTypeGetName()
/external/tensorflow/tensorflow/lite/kernels/
Delementwise.cc28 bool IsNumericSupportedType(const TfLiteType type) { in IsNumericSupportedType()
32 bool IsLogicalSupportedType(const TfLiteType type) { in IsLogicalSupportedType()
36 typedef bool (*IsSupportedType)(TfLiteType);
55 T func(T), TfLiteType expected_type) { in EvalImpl()
/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil.h39 TF_DataType GetTensorFlowDataType(TfLiteType type);
42 TfLiteType GetTensorFlowLiteType(TF_DataType);
Dutil.cc57 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType()
82 TfLiteType GetTensorFlowLiteType(TF_DataType type) { in GetTensorFlowLiteType()
Dtest_util.h81 TfLiteType GetType(int tensor_index);
89 const std::vector<int>& outputs, TfLiteType type,
Dtest_util.cc66 TfLiteType FlexModelTest::GetType(int tensor_index) { in GetType()
71 const std::vector<int>& outputs, TfLiteType type, in AddTensors()
/external/tensorflow/tensorflow/lite/python/interpreter_wrapper/
Dpython_utils.h30 int TfLiteTypeToPyArrayType(TfLiteType tf_lite_type);
32 TfLiteType TfLiteTypeFromPyArray(PyArrayObject* array);
Dpython_utils.cc31 int TfLiteTypeToPyArrayType(TfLiteType tf_lite_type) { in TfLiteTypeToPyArrayType()
58 TfLiteType TfLiteTypeFromPyArray(PyArrayObject* array) { in TfLiteTypeFromPyArray()
/external/tensorflow/tensorflow/lite/core/
Dsubgraph.h89 int tensor_index, TfLiteType type, const char* name,
98 int tensor_index, TfLiteType type, const char* name, const size_t rank,
107 int tensor_index, TfLiteType type, const char* name,
113 TfLiteStatus SetTensorParametersReadWrite(int tensor_index, TfLiteType type,
358 TfLiteStatus BytesRequired(TfLiteType type, const int* dims, size_t dims_size,
/external/tensorflow/tensorflow/lite/java/src/main/native/
Dtensor_jni.cc63 size_t ElementByteSize(TfLiteType data_type) { in ElementByteSize()
92 size_t WriteOneDimensionalArray(JNIEnv* env, jobject object, TfLiteType type, in WriteOneDimensionalArray()
141 size_t ReadOneDimensionalArray(JNIEnv* env, TfLiteType data_type, in ReadOneDimensionalArray()
185 size_t ReadMultiDimensionalArray(JNIEnv* env, TfLiteType data_type, char* src, in ReadMultiDimensionalArray()
237 size_t WriteMultiDimensionalArray(JNIEnv* env, jobject src, TfLiteType type, in WriteMultiDimensionalArray()
/external/tensorflow/tensorflow/lite/delegates/nnapi/
Dnnapi_delegate.cc55 bool IsFloat(TfLiteType type) { in IsFloat()
64 bool IsQuantized(TfLiteType type) { in IsQuantized()
82 const TfLiteType input_type = context->tensors[input_id].type; in IsHybridOperator()
83 const TfLiteType filter_type = context->tensors[filter_id].type; in IsHybridOperator()
207 int DequantizedAnnIndex(int ann_index, TfLiteType type) const { in DequantizedAnnIndex()
216 void Add(int ann_index, TfLiteType type, int dequantized_ann_index) { in Add()
225 std::vector<std::tuple<int, TfLiteType, int>> mapping_;
299 TfLiteType dequantized_type) { in AddDequantize()
419 TfLiteType tensor_type = tensor->type; in AddTensor()
1206 const TfLiteType type = context->tensors[tensor_id].type; in AddDequantizeOperatorsWhereNeeded()
/external/tensorflow/tensorflow/lite/experimental/micro/
Dsimple_tensor_allocator.cc23 TfLiteStatus TfLiteTypeSizeOf(TfLiteType type, size_t* size, in TfLiteTypeSizeOf()
59 TfLiteType tf_lite_type; in BytesRequired()
/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dfully_connected.cc50 TfLiteType data_type, const TfLiteTensor* input, in CalculateOpData()
151 TfLiteType data_type = input->type; in Eval()
Ddepthwise_conv.cc54 const TfLiteType data_type, OpData* data) { in CalculateOpData()
166 const TfLiteType data_type = input->type; in Eval()
/external/tensorflow/tensorflow/lite/tools/accuracy/
Dutils.h37 DataType GetTFDataType(TfLiteType tflite_type);
Dutils.cc35 DataType GetTFDataType(TfLiteType tflite_type) { in GetTFDataType()
/external/tensorflow/tensorflow/lite/core/api/
Dflatbuffer_conversions.h63 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type,
Dflatbuffer_conversions.cc56 TfLiteStatus ConvertTensorType(TensorType tensor_type, TfLiteType* type, in ConvertTensorType()
669 ? TfLiteType::kTfLiteInt64 in ParseOpData()
670 : TfLiteType::kTfLiteInt32; in ParseOpData()
/external/tensorflow/tensorflow/lite/experimental/writer/
Denum_mapping.h58 inline TensorType TfLiteTypeToSchemaType(TfLiteType type) { in TfLiteTypeToSchemaType()
/external/tensorflow/tensorflow/lite/experimental/c/
Dc_api.h60 typedef TfLiteType TFL_Type;

12