Home
last modified time | relevance | path

Searched refs:MeDataType (Results 1 – 4 of 4) 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()
64 static std::map<MeDataType, size_t> datatype_size_map = {
65 …{MeDataType::kNumberTypeFloat16, sizeof(float) / 2}, {MeDataType::kNumberTypeFloat32, sizeof(float…
[all …]
Dutil.h50 static GeDataType ConvertDataType(const MeDataType &type);
66 static size_t GetDataTypeSize(const MeDataType &type);
76 …static std::shared_ptr<GeTensorDesc> GetGeTensorDesc(const ShapeVector &shape, const MeDataType &m…
146 static MeDataType ConvertGeDataType(const GeDataType &type);
Dtypes.h43 using MeDataType = mindspore::TypeId; variable
/third_party/mindspore/tests/ut/cpp/transform/
Dconvert_test.cc664 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()
683 ASSERT_EQ(TransformUtil::GetDataTypeSize(MeDataType::kNumberTypeFloat32), 4); in TEST_F()
[all …]