Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil_test.cc128 EXPECT_EQ(kTfLiteFloat16, GetTensorFlowLiteType(TF_HALF)); in TEST()
129 EXPECT_EQ(kTfLiteFloat32, GetTensorFlowLiteType(TF_FLOAT)); in TEST()
130 EXPECT_EQ(kTfLiteFloat64, GetTensorFlowLiteType(TF_DOUBLE)); in TEST()
131 EXPECT_EQ(kTfLiteInt16, GetTensorFlowLiteType(TF_INT16)); in TEST()
132 EXPECT_EQ(kTfLiteInt32, GetTensorFlowLiteType(TF_INT32)); in TEST()
133 EXPECT_EQ(kTfLiteUInt8, GetTensorFlowLiteType(TF_UINT8)); in TEST()
134 EXPECT_EQ(kTfLiteInt64, GetTensorFlowLiteType(TF_INT64)); in TEST()
135 EXPECT_EQ(kTfLiteUInt64, GetTensorFlowLiteType(TF_UINT64)); in TEST()
136 EXPECT_EQ(kTfLiteComplex64, GetTensorFlowLiteType(TF_COMPLEX64)); in TEST()
137 EXPECT_EQ(kTfLiteComplex128, GetTensorFlowLiteType(TF_COMPLEX128)); in TEST()
[all …]
Dutil.cc32 tensor->type = GetTensorFlowLiteType(static_cast<TF_DataType>(src.dtype())); in CopyShapeAndType()
96 TfLiteType GetTensorFlowLiteType(TF_DataType type) { in GetTensorFlowLiteType() function
Dutil.h42 TfLiteType GetTensorFlowLiteType(TF_DataType);