Home
last modified time | relevance | path

Searched refs:GetTensorFlowLiteType (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/flex/
H A Dutil_test.cc135 EXPECT_EQ(kTfLiteFloat16, GetTensorFlowLiteType(TF_HALF)); in TEST()
136 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
137 EXPECT_EQ(kTfLiteFloat64, GetTensorFlowLiteType(TF_DOUBLE)); in TEST()
138 EXPECT_EQ(kTfLiteInt16, GetTensorFlowLiteType(TF_INT16)); in TEST()
139 EXPECT_EQ(kTfLiteInt32, GetTensorFlowLiteType(TF_INT32)); in TEST()
140 EXPECT_EQ(kTfLiteUInt8, GetTensorFlowLiteType(TF_UINT8)); in TEST()
141 EXPECT_EQ(kTfLiteInt64, GetTensorFlowLiteType(TF_INT64)); in TEST()
142 EXPECT_EQ(kTfLiteUInt64, GetTensorFlowLiteType(TF_UINT64)); in TEST()
143 EXPECT_EQ(kTfLiteComplex64, GetTensorFlowLiteType(TF_COMPLEX64)); in TEST()
144 EXPECT_EQ(kTfLiteComplex128, GetTensorFlowLiteType(TF_COMPLEX128)); in TEST()
[all …]
H A Dutil.cc44 tensor->type = GetTensorFlowLiteType(static_cast<TF_DataType>(src.dtype())); in CopyShapeAndType()
110 TfLiteType GetTensorFlowLiteType(TF_DataType type) { in GetTensorFlowLiteType() function
H A Dutil.h46 TfLiteType GetTensorFlowLiteType(TF_DataType);