/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() 35 tensor::TensorPtr TensorConstructUtils::CreateOnesTensor(const TypePtr &type_ptr, const std::vector… in CreateOnesTensor() 76 tensor::TensorPtr TensorConstructUtils::CreateTensor(const TypePtr &type_ptr, const std::vector<int… in CreateTensor() 84 TypeId TensorConstructUtils::ExtractTypeId(const TypePtr &type_ptr) { in ExtractTypeId()
|
/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() 427 .def(py::init([](const Tensor &tensor, const TypePtr &type_ptr) { in __anonde8b974e0102() 435 .def(py::init([](const TypePtr &type_ptr, const py::tuple &shape) { in __anonde8b974e0102() 440 .def(py::init([](const TypePtr &type_ptr, const py::list &shape) { in __anonde8b974e0102() 445 .def(py::init([](const py::array &input, const TypePtr &type_ptr) { in __anonde8b974e0102() 449 .def(py::init([](const py::float_ input, const TypePtr &type_ptr) { in __anonde8b974e0102() 453 .def(py::init([](const py::int_ input, const TypePtr &type_ptr) { in __anonde8b974e0102() 457 .def(py::init([](const py::list &input, const TypePtr &type_ptr) { in __anonde8b974e0102() 461 .def(py::init([](const py::tuple &input, const TypePtr &type_ptr) { in __anonde8b974e0102() 468 .def(py::init([](const py::object &input, const TypePtr &type_ptr) { in __anonde8b974e0102()
|
/third_party/mindspore/mindspore/core/ir/ |
D | meta_tensor.cc | 32 MetaTensor::MetaTensor(const TypePtr &type_ptr, const ShapeVector &shape) { in MetaTensor() 76 TypePtr MetaTensor::SetDtype(const TypePtr type_ptr) { in SetDtype()
|
/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 65 TypePtr type_ptr = TypeIdToType(AnfAlgo::GetOutputDeviceDataType(anf_node, i)); in Init() local
|
/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
|
/third_party/mindspore/mindspore/lite/tools/converter/acl/mapper/ |
D | cast_mapper.cc | 45 TypePtr type_ptr = TypeIdToType(TypeId(dst_type)); in Mapper() local
|
/third_party/mindspore/tests/ut/cpp/ops/ |
D | test_ops_unstack.cc | 57 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_custom_predict.cc | 58 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_unpack.cc | 57 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_hashtable_lookup.cc | 59 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_layer_norm_beta_gamma_backprop_v2.cc | 61 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_topk.cc | 70 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_merge.cc | 60 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_layer_norm_x_backprop_v2.cc | 68 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_detection_post_process.cc | 66 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_batch_norm.cc | 68 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
D | test_ops_batch_norm_grad.cc | 71 auto type_ptr = abstract->BuildType(); in TEST_F() local
|
/third_party/mindspore/mindspore/core/ops/ |
D | arg_min.cc | 33 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
|
D | arg_max.cc | 59 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
|
/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
|
D | apply_momentum_weight_scale_fusion.cc | 47 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
|
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/ |
D | utils.cc | 116 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in GetTypeFromNode() local
|
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/ |
D | multitype_funcgraph.cc | 65 TypePtr type_ptr = nullptr; in PyRegister() local
|
/third_party/mindspore/mindspore/core/ir/dtype/ |
D | type.cc | 168 size_t GetTypeByte(const TypePtr &type_ptr) { in GetTypeByte()
|
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/ |
D | aicpu_kernel_build.cc | 66 auto type_ptr = TypeIdToType(AnfAlgo::GetInputDeviceDataType(anf_node, i)); in SetIOIputSize() local 94 TypePtr type_ptr = TypeIdToType(AnfAlgo::GetOutputDeviceDataType(anf_node, i)); in SetIOSize() local
|