Searched refs:GetDataTypeAttr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 112 tensorflow::DataType GetDataTypeAttr(const NodeDef& node, in GetDataTypeAttr() function
162 const tensorflow::DataType& value = GetDataTypeAttr(node, attr_name); in CheckOptionalAttr()
655 return GetDataTypeAttr(node, a.type_attr()); in GetOutputTypesFromNodeDef()
737 const auto& output_type = GetDataTypeAttr(node, "Tout"); in ConvertUnsupportedOperator()
778 const auto dtype = GetDataTypeAttr(node, "dtype"); in ConvertConstOperator()
921 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertDepthwiseConvOperator()
994 tensorflow::DataType dtype = GetDataTypeAttr(node, "T"); in ConvertDepthToSpaceOperator()
1017 tensorflow::DataType dtype = GetDataTypeAttr(node, "T"); in ConvertSpaceToDepthOperator()
1042 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertBiasAddOperator()
1057 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_INT32); in ConvertRandomUniform()
[all …]
|