Searched refs:me_type (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/tests/ut/python/ir/ |
| D | test_dtype.py | 84 me_type = dtype.get_py_obj_dtype(x)
85 assert me_type == ms.float64
86 me_type = dtype.get_py_obj_dtype(type(x))
87 assert me_type == ms.float64
90 me_type = dtype.get_py_obj_dtype(type(x))
91 assert me_type == ms.int64
92 me_type = dtype.get_py_obj_dtype(x)
93 assert me_type == ms.int64
96 me_type = dtype.get_py_obj_dtype(type(x))
97 assert me_type == ms.bool_
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/transform/graph_ir/ |
| D | op_adapter.cc | 325 TypeId me_type = type->type_id(); in CreateOutputDesc() local
326 if (kObjectTypeTensorType == me_type) { in CreateOutputDesc()
327 me_type = dyn_cast<TensorType>(type)->element()->type_id(); in CreateOutputDesc()
330 …til::GetGeTensorDesc((shape_ptr == nullptr) ? ShapeVector{} : shape_ptr->shape(), me_type, format); in CreateOutputDesc()
382 TypeId me_type = node->Type()->type_id(); in CreateNodeDesc() local
383 if (kObjectTypeTensorType == me_type) { in CreateNodeDesc()
384 me_type = dyn_cast<TensorType>(node->Type())->element()->type_id(); in CreateNodeDesc()
386 if (me_type <= kNumberTypeBegin || me_type >= kNumberTypeEnd) { in CreateNodeDesc()
396 auto desc = TransformUtil::GetGeTensorDesc(shape, me_type, format); in CreateNodeDesc()
|
| D | util.h | 76 … std::shared_ptr<GeTensorDesc> GetGeTensorDesc(const ShapeVector &shape, const MeDataType &me_type,
139 … GenerateMeTensor(const GeTensorPtr &ge_tensor, const ShapeVector &me_dims, const TypeId &me_type);
|
| D | op_adapter_util.cc | 124 TypeId me_type = type->type_id(); in ConvertAnyUtil() local
125 if (kObjectTypeTensorType == me_type) { in ConvertAnyUtil()
126 me_type = dyn_cast<TensorType>(type)->element()->type_id(); in ConvertAnyUtil()
128 return TransformUtil::ConvertDataType(me_type); in ConvertAnyUtil()
|
| D | util.cc | 108 …eTensorDesc> TransformUtil::GetGeTensorDesc(const ShapeVector &me_shape, const MeDataType &me_type, in GetGeTensorDesc() argument
130 GeDataType data_type = ConvertDataType(me_type); in GetGeTensorDesc()
132 MS_LOG(ERROR) << "undefined data type :" << me_type; in GetGeTensorDesc()
345 const TypeId &me_type) { in GenerateMeTensor() argument
346 MeTensor me_tensor(me_type, me_dims); in GenerateMeTensor()
|
| D | convert.cc | 966 TypeId me_type = node->Type()->type_id(); in UpdateDataOpDesc() local
967 if (kObjectTypeTensorType == me_type) { in UpdateDataOpDesc()
968 me_type = dyn_cast<TensorType>(node->Type())->element()->type_id(); in UpdateDataOpDesc()
972 MS_LOG(INFO) << "input shape is " << buf.str() << ", type is " << me_type; in UpdateDataOpDesc()
983 auto desc = TransformUtil::GetGeTensorDesc(shape, me_type, format); in UpdateDataOpDesc()
|