Searched refs:GeDataType (Results 1 – 8 of 8) sorted by relevance
/third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
D | util.cc | 47 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 …]
|
D | util.h | 50 static GeDataType ConvertDataType(const MeDataType &type); 146 static MeDataType ConvertGeDataType(const GeDataType &type);
|
D | types.h | 44 using GeDataType = ge::DataType; variable
|
D | op_adapter_util.h | 44 GeDataType ConvertAnyUtil(const ValuePtr &value, const AnyTraits<GEType>);
|
D | op_adapter_util.cc | 116 GeDataType ConvertAnyUtil(const ValuePtr &value, const AnyTraits<GEType>) { in ConvertAnyUtil()
|
D | op_adapter.h | 417 static GeDataType ConvertAny(const ValuePtr &value, const AnyTraits<GEType> anyTraitsGE) { in ConvertAny()
|
/third_party/mindspore/tests/ut/cpp/transform/ |
D | convert_test.cc | 637 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/ |
D | framework_parser.py | 67 class GeDataType(enum.IntEnum): class 566 attr_value = GeDataType.get_data_type_name(
|