| /third_party/mindspore/mindspore/core/utils/ |
| D | tensor_construct_utils.cc | 22 tensor::TensorPtr TensorConstructUtils::CreateZerosTensor(const TypePtr &type_ptr, const std::vecto… in CreateZerosTensor() argument
23 MS_EXCEPTION_IF_NULL(type_ptr); in CreateZerosTensor()
24 auto type_id = ExtractTypeId(type_ptr); in CreateZerosTensor()
35 tensor::TensorPtr TensorConstructUtils::CreateOnesTensor(const TypePtr &type_ptr, const std::vector… in CreateOnesTensor() argument
36 MS_EXCEPTION_IF_NULL(type_ptr); in CreateOnesTensor()
37 auto type_id = ExtractTypeId(type_ptr); in CreateOnesTensor()
76 tensor::TensorPtr TensorConstructUtils::CreateTensor(const TypePtr &type_ptr, const std::vector<int… in CreateTensor() argument
78 MS_EXCEPTION_IF_NULL(type_ptr); in CreateTensor()
79 auto type_id = ExtractTypeId(type_ptr); in CreateTensor()
84 TypeId TensorConstructUtils::ExtractTypeId(const TypePtr &type_ptr) { in ExtractTypeId() argument
[all …]
|
| /third_party/mindspore/mindspore/core/ir/ |
| D | meta_tensor.cc | 32 MetaTensor::MetaTensor(const TypePtr &type_ptr, const ShapeVector &shape) { in MetaTensor() argument
34 if (type_ptr != nullptr) { in MetaTensor()
35 data_type = type_ptr->type_id(); in MetaTensor()
76 TypePtr MetaTensor::SetDtype(const TypePtr type_ptr) { in SetDtype() argument
77 if (type_ptr == nullptr) { in SetDtype()
81 (void)set_data_type(type_ptr->type_id()); in SetDtype()
82 return type_ptr; in SetDtype()
|
| /third_party/mindspore/mindspore/ccsrc/pybind_api/ir/ |
| D | tensor_py.cc | 197 TensorPtr TensorPy::MakeTensor(const py::array &input, const TypePtr &type_ptr) { in MakeTensor() argument
201 auto data_type = type_ptr ? type_ptr->type_id() : TypeId::kTypeUnknown; in MakeTensor()
427 .def(py::init([](const Tensor &tensor, const TypePtr &type_ptr) { in __anonde8b974e0102() argument
428 TypeId data_type = type_ptr ? type_ptr->type_id() : kTypeUnknown; in __anonde8b974e0102()
435 .def(py::init([](const TypePtr &type_ptr, const py::tuple &shape) { in __anonde8b974e0102() argument
436 … auto data_type = type_ptr ? type_ptr->type_id() : TypeId::kNumberTypeFloat64; in __anonde8b974e0102()
440 .def(py::init([](const TypePtr &type_ptr, const py::list &shape) { in __anonde8b974e0102() argument
441 … auto data_type = type_ptr ? type_ptr->type_id() : TypeId::kNumberTypeFloat64; in __anonde8b974e0102()
445 .def(py::init([](const py::array &input, const TypePtr &type_ptr) { in __anonde8b974e0102() argument
446 return TensorPy::MakeTensor(input, type_ptr); in __anonde8b974e0102()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/host/ |
| D | host_kernel_mod.cc | 55 TypePtr type_ptr = TypeIdToType(AnfAlgo::GetInputDeviceDataType(anf_node, i)); in Init() local
57 if (!GetShapeSize(shape_i, type_ptr, &size_i)) { in Init()
65 TypePtr type_ptr = TypeIdToType(AnfAlgo::GetOutputDeviceDataType(anf_node, i)); in Init() local
66 MS_EXCEPTION_IF_NULL(type_ptr); in Init()
68 if (!GetShapeSize(shape_i, type_ptr, &size_i)) { in Init()
|
| /third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
| D | cast_mapper.cc | 45 TypePtr type_ptr = TypeIdToType(TypeId(dst_type)); in Mapper() local
46 if (type_ptr == nullptr) { in Mapper()
50 ValueNodePtr value_node = NewValueNode(type_ptr); in Mapper()
|
| /third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
| D | multitype_funcgraph.cc | 65 TypePtr type_ptr = nullptr; in PyRegister() local
68 type_ptr = StringToType(type_name); in PyRegister()
69 if (type_ptr == nullptr) { in PyRegister()
73 type_ptr = type_in.cast<TypePtr>(); in PyRegister()
77 types.push_back(type_ptr); in PyRegister()
|
| /third_party/mindspore/tests/ut/cpp/ops/ |
| D | test_ops_unstack.cc | 57 auto type_ptr = abstract->BuildType(); in TEST_F() local
58 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
59 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_unpack.cc | 57 auto type_ptr = abstract->BuildType(); in TEST_F() local
58 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
59 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_custom_predict.cc | 58 auto type_ptr = abstract->BuildType(); in TEST_F() local
59 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
60 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_hashtable_lookup.cc | 59 auto type_ptr = abstract->BuildType(); in TEST_F() local
60 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
61 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_merge.cc | 60 auto type_ptr = abstract->BuildType(); in TEST_F() local
61 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
62 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_layer_norm_beta_gamma_backprop_v2.cc | 61 auto type_ptr = abstract->BuildType(); in TEST_F() local
62 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
63 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_topk.cc | 70 auto type_ptr = abstract->BuildType(); in TEST_F() local
71 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
72 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_layer_norm_x_backprop_v2.cc | 68 auto type_ptr = abstract->BuildType(); in TEST_F() local
69 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
70 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_detection_post_process.cc | 66 auto type_ptr = abstract->BuildType(); in TEST_F() local
67 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
68 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_batch_norm.cc | 68 auto type_ptr = abstract->BuildType(); in TEST_F() local
69 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
70 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| D | test_ops_batch_norm_grad.cc | 71 auto type_ptr = abstract->BuildType(); in TEST_F() local
72 MS_EXCEPTION_IF_NULL(type_ptr); in TEST_F()
73 auto type = type_ptr->cast<TuplePtr>(); in TEST_F()
|
| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/ |
| D | apply_momentum_scale_fusion.cc | 44 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
45 MS_EXCEPTION_IF_NULL(type_ptr); in IsScalar()
46 auto element = type_ptr->element(); in IsScalar()
|
| D | apply_momentum_weight_scale_fusion.cc | 47 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
48 MS_EXCEPTION_IF_NULL(type_ptr); in IsScalar()
49 auto element = type_ptr->element(); in IsScalar()
|
| /third_party/mindspore/mindspore/core/ir/dtype/ |
| D | type.cc | 168 size_t GetTypeByte(const TypePtr &type_ptr) { in GetTypeByte() argument
169 if (type_ptr && type_ptr->isa<Number>()) { in GetTypeByte()
170 auto number = dyn_cast<Number>(type_ptr); in GetTypeByte()
|
| /third_party/mindspore/mindspore/core/ops/ |
| D | arg_min.cc | 33 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
34 return type_ptr->type_id(); in get_output_type()
|
| D | arg_max.cc | 59 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
60 return type_ptr->type_id(); in get_output_type()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/hccl/ |
| D | hcom_util.cc | 74 TypeId type_ptr; in GetHcomDataType() local
76 type_ptr = AnfAlgo::GetOutputDeviceDataType(anf_node, i); in GetHcomDataType()
78 type_ptr = AnfAlgo::GetInputDeviceDataType(anf_node, i); in GetHcomDataType()
80 data_type_list->emplace_back(ConvertHcclType(type_ptr)); in GetHcomDataType()
|
| /third_party/mindspore/mindspore/lite/test/common/ |
| D | import_from_meta_graphT.cc | 52 auto type_ptr = TypeIdToType(type_id); in ConverterConstTensor() local
97 auto type_ptr = TypeIdToType(type_id); in ConvertTensorToAbstractTensor() local
101 auto ptr = std::make_shared<abstract::AbstractTensor>(type_ptr, shape_vector); in ConvertTensorToAbstractTensor()
|
| /third_party/mindspore/mindspore/core/ops/grad/ |
| D | group_conv2d_grad_input.cc | 176 auto type_ptr = input_args[0]->BuildType(); in GroupConv2DGradInputInfer() local
177 MS_EXCEPTION_IF_NULL(type_ptr); in GroupConv2DGradInputInfer()
178 auto type_tensor_ptr = type_ptr->cast<TensorTypePtr>(); in GroupConv2DGradInputInfer()
|