Home
last modified time | relevance | path

Searched defs:type_ptr (Results 1 – 25 of 47) sorted by relevance

12

/third_party/mindspore/mindspore/core/utils/
Dtensor_construct_utils.cc22 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/
Dtensor_py.cc197 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/
Dmeta_tensor.cc32 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/
Dhost_kernel_mod.cc55 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/
Dimport_from_meta_graphT.cc52 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/
Dcast_mapper.cc45 TypePtr type_ptr = TypeIdToType(TypeId(dst_type)); in Mapper() local
/third_party/mindspore/tests/ut/cpp/ops/
Dtest_ops_unstack.cc57 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_custom_predict.cc58 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_unpack.cc57 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_hashtable_lookup.cc59 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_layer_norm_beta_gamma_backprop_v2.cc61 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_topk.cc70 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_merge.cc60 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_layer_norm_x_backprop_v2.cc68 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_detection_post_process.cc66 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_batch_norm.cc68 auto type_ptr = abstract->BuildType(); in TEST_F() local
Dtest_ops_batch_norm_grad.cc71 auto type_ptr = abstract->BuildType(); in TEST_F() local
/third_party/mindspore/mindspore/core/ops/
Darg_min.cc33 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
Darg_max.cc59 auto type_ptr = GetAttr(kOutputType)->cast<TensorTypePtr>()->element(); in get_output_type() local
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/gpu/
Dapply_momentum_scale_fusion.cc44 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
Dapply_momentum_weight_scale_fusion.cc47 auto type_ptr = dyn_cast<TensorType>(dtype); in IsScalar() local
/third_party/mindspore/mindspore/lite/tools/converter/acl/common/
Dutils.cc116 auto type_ptr = abstract_tensor->element()->GetTypeTrack(); in GetTypeFromNode() local
/third_party/mindspore/mindspore/ccsrc/frontend/operator/composite/
Dmultitype_funcgraph.cc65 TypePtr type_ptr = nullptr; in PyRegister() local
/third_party/mindspore/mindspore/core/ir/dtype/
Dtype.cc168 size_t GetTypeByte(const TypePtr &type_ptr) { in GetTypeByte()
/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/aicpu/
Daicpu_kernel_build.cc66 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

12