Searched refs:GetTensorFlowDataType (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/ |
| H A D | util_test.cc | 117 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteNoType)); in TEST()
118 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
119 EXPECT_EQ(TF_HALF, GetTensorFlowDataType(kTfLiteFloat16)); in TEST()
120 EXPECT_EQ(TF_DOUBLE, GetTensorFlowDataType(kTfLiteFloat64)); in TEST()
121 EXPECT_EQ(TF_INT16, GetTensorFlowDataType(kTfLiteInt16)); in TEST()
122 EXPECT_EQ(TF_INT32, GetTensorFlowDataType(kTfLiteInt32)); in TEST()
123 EXPECT_EQ(TF_UINT8, GetTensorFlowDataType(kTfLiteUInt8)); in TEST()
124 EXPECT_EQ(TF_INT64, GetTensorFlowDataType(kTfLiteInt64)); in TEST()
125 EXPECT_EQ(TF_UINT64, GetTensorFlowDataType(kTfLiteUInt64)); in TEST()
126 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
[all …]
|
| H A D | util.cc | 69 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType() function
231 tensorflow::DataType(GetTensorFlowDataType(tflite_tensor->type)), shape); in CreateTfTensorFromTfLiteTensor()
|
| H A D | delegate_data.cc | 98 TF_DataType dtype = tflite::flex::GetTensorFlowDataType( in BuildFunctionDefProto()
103 TF_DataType dtype = tflite::flex::GetTensorFlowDataType( in BuildFunctionDefProto()
|
| H A D | util.h | 43 TF_DataType GetTensorFlowDataType(TfLiteType type);
|
| H A D | buffer_map_util.cc | 187 GetTensorFlowDataType(tensor->type), shape, buf); in SetTfTensorFromTfLite()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | export_tensorflow.cc | 53 tensorflow::DataType GetTensorFlowDataType(ArrayDataType data_type, in GetTensorFlowDataType() function
86 return GetTensorFlowDataType(data_type, "op '" + op_name + "'"); in GetTensorFlowDataTypeForOp()
89 tensorflow::DataType GetTensorFlowDataType(const Model& model, in GetTensorFlowDataType() function
91 return GetTensorFlowDataType(model.GetArray(array_name).data_type, in GetTensorFlowDataType()
624 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
646 GetTensorFlowDataType(model, src_op.inputs[1])); in ConvertFullyConnectedOperator()
655 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
668 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
694 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddOperator()
707 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddNOperator()
[all …]
|