Home
last modified time | relevance | path

Searched refs:GetTensorFlowDataType (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/lite/delegates/flex/
Dutil_test.cc110 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteNoType)); in TEST()
111 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
112 EXPECT_EQ(TF_HALF, GetTensorFlowDataType(kTfLiteFloat16)); in TEST()
113 EXPECT_EQ(TF_DOUBLE, GetTensorFlowDataType(kTfLiteFloat64)); in TEST()
114 EXPECT_EQ(TF_INT16, GetTensorFlowDataType(kTfLiteInt16)); in TEST()
115 EXPECT_EQ(TF_INT32, GetTensorFlowDataType(kTfLiteInt32)); in TEST()
116 EXPECT_EQ(TF_UINT8, GetTensorFlowDataType(kTfLiteUInt8)); in TEST()
117 EXPECT_EQ(TF_INT64, GetTensorFlowDataType(kTfLiteInt64)); in TEST()
118 EXPECT_EQ(TF_UINT64, GetTensorFlowDataType(kTfLiteUInt64)); in TEST()
119 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
[all …]
Dutil.h39 TF_DataType GetTensorFlowDataType(TfLiteType type);
Dutil.cc57 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType() function
Dbuffer_map.cc174 GetTensorFlowDataType(tensor->type), shape, buf); in SetFromTfLite()
/external/tensorflow/tensorflow/lite/toco/
Dexport_tensorflow.cc52 tensorflow::DataType GetTensorFlowDataType(ArrayDataType data_type, in GetTensorFlowDataType() function
81 return GetTensorFlowDataType(data_type, "op '" + op_name + "'"); in GetTensorFlowDataTypeForOp()
84 tensorflow::DataType GetTensorFlowDataType(const Model& model, in GetTensorFlowDataType() function
86 return GetTensorFlowDataType(model.GetArray(array_name).data_type, in GetTensorFlowDataType()
619 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
641 GetTensorFlowDataType(model, src_op.inputs[1])); in ConvertFullyConnectedOperator()
650 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
663 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
689 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddOperator()
702 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddNOperator()
[all …]