Home
last modified time | relevance | path

Searched refs:GeDataType (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/
Dutil.cc47 static std::map<MeDataType, GeDataType> datatype_trans_map = {
48 …{MeDataType::kNumberTypeFloat16, GeDataType::DT_FLOAT16}, {MeDataType::kNumberTypeFloat32, GeDataT…
49 …{MeDataType::kNumberTypeFloat64, GeDataType::DT_DOUBLE}, {MeDataType::kNumberTypeInt8, GeDataType
50 …{MeDataType::kNumberTypeInt16, GeDataType::DT_INT16}, {MeDataType::kNumberTypeInt32, GeDataTyp…
51 …{MeDataType::kNumberTypeInt64, GeDataType::DT_INT64}, {MeDataType::kNumberTypeUInt8, GeDataTyp…
52 …{MeDataType::kNumberTypeUInt16, GeDataType::DT_UINT16}, {MeDataType::kNumberTypeUInt32, GeDataTy…
53 …{MeDataType::kNumberTypeUInt64, GeDataType::DT_UINT64}, {MeDataType::kNumberTypeBool, GeDataType
55 GeDataType TransformUtil::ConvertDataType(const MeDataType &type) { in ConvertDataType()
60 return GeDataType::DT_UNDEFINED; in ConvertDataType()
130 GeDataType data_type = ConvertDataType(me_type); in GetGeTensorDesc()
[all …]
Dutil.h50 static GeDataType ConvertDataType(const MeDataType &type);
146 static MeDataType ConvertGeDataType(const GeDataType &type);
Dtypes.h44 using GeDataType = ge::DataType; variable
Dop_adapter_util.h44 GeDataType ConvertAnyUtil(const ValuePtr &value, const AnyTraits<GEType>);
Dop_adapter_util.cc116 GeDataType ConvertAnyUtil(const ValuePtr &value, const AnyTraits<GEType>) { in ConvertAnyUtil()
Dop_adapter.h417 static GeDataType ConvertAny(const ValuePtr &value, const AnyTraits<GEType> anyTraitsGE) { in ConvertAny()
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc637 ASSERT_EQ(ge_tensor_ptr->GetTensorDesc().GetDataType(), GeDataType::DT_FLOAT); in TEST_F()
664 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeFloat16), GeDataType::DT_FLOAT16); in TEST_F()
665 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeFloat32), GeDataType::DT_FLOAT); in TEST_F()
666 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeFloat64), GeDataType::DT_DOUBLE); in TEST_F()
667 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeInt8), GeDataType::DT_INT8); in TEST_F()
668 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeInt16), GeDataType::DT_INT16); in TEST_F()
669 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeInt32), GeDataType::DT_INT32); in TEST_F()
670 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeInt64), GeDataType::DT_INT64); in TEST_F()
671 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeUInt32), GeDataType::DT_UINT32); in TEST_F()
672 ASSERT_EQ(TransformUtil::ConvertDataType(MeDataType::kNumberTypeBool), GeDataType::DT_BOOL); in TEST_F()
/third_party/mindspore/mindspore/profiler/parser/
Dframework_parser.py67 class GeDataType(enum.IntEnum): class
566 attr_value = GeDataType.get_data_type_name(