Searched refs:GetTensorFlowDataType (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | util_test.cc | 110 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteNoType)); in TEST()
111 EXPECT_EQ(TF_FLOAT, GetTensorFlowDataType(kTfLiteFloat32)); in TEST()
112 EXPECT_EQ(TF_INT16, GetTensorFlowDataType(kTfLiteInt16)); in TEST()
113 EXPECT_EQ(TF_INT32, GetTensorFlowDataType(kTfLiteInt32)); in TEST()
114 EXPECT_EQ(TF_UINT8, GetTensorFlowDataType(kTfLiteUInt8)); in TEST()
115 EXPECT_EQ(TF_INT64, GetTensorFlowDataType(kTfLiteInt64)); in TEST()
116 EXPECT_EQ(TF_COMPLEX64, GetTensorFlowDataType(kTfLiteComplex64)); in TEST()
117 EXPECT_EQ(TF_STRING, GetTensorFlowDataType(kTfLiteString)); in TEST()
118 EXPECT_EQ(TF_BOOL, GetTensorFlowDataType(kTfLiteBool)); in TEST()
|
| D | util.h | 39 TF_DataType GetTensorFlowDataType(TfLiteType type);
|
| D | util.cc | 57 TF_DataType GetTensorFlowDataType(TfLiteType type) { in GetTensorFlowDataType() function
|
| D | buffer_map.cc | 164 GetTensorFlowDataType(tensor->type), shape, buf); in SetFromTfLite()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | export_tensorflow.cc | 51 tensorflow::DataType GetTensorFlowDataType(ArrayDataType data_type, in GetTensorFlowDataType() function
78 return GetTensorFlowDataType(data_type, "op '" + op_name + "'"); in GetTensorFlowDataTypeForOp()
81 tensorflow::DataType GetTensorFlowDataType(const Model& model, in GetTensorFlowDataType() function
83 return GetTensorFlowDataType(model.GetArray(array_name).data_type, in GetTensorFlowDataType()
614 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
636 GetTensorFlowDataType(model, src_op.inputs[1])); in ConvertFullyConnectedOperator()
645 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
658 GetTensorFlowDataType(model, src_op.inputs[0])); in ConvertFullyConnectedOperator()
684 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddOperator()
697 GetTensorFlowDataType(model, src_op.outputs[0])); in ConvertAddNOperator()
[all …]
|