HomeSort by relevance Sort by last modified time
    Searched refs:GetTensorFlowDataType (Results 1 - 5 of 5) sorted by null

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

Completed in 86 milliseconds