Searched refs:GetDataTypeAttr (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/ |
| D | import_tensorflow.cc | 110 tensorflow::DataType GetDataTypeAttr(const NodeDef& node, in GetDataTypeAttr() function
159 const tensorflow::DataType& value = GetDataTypeAttr(node, attr_name); in CheckOptionalAttr()
570 const auto dtype = GetDataTypeAttr(node, "dtype"); in ConvertConstOperator()
703 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertDepthwiseConvOperator()
769 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertDepthToSpaceOperator()
785 tensorflow::DataType dtype = GetDataTypeAttr(node, "T"); in ConvertSpaceToDepthOperator()
810 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertBiasAddOperator()
825 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_INT32); in ConvertRandomUniform()
829 op->dtype = ConvertDataType(GetDataTypeAttr(node, "dtype")); in ConvertRandomUniform()
1020 CHECK_EQ(GetDataTypeAttr(node, "T"), DT_FLOAT); in ConvertMaxPoolOperator()
[all …]
|