| /third_party/mindspore/mindspore-src/source/tests/st/map_parameter/ |
| D | test_map_parameter.py | 62 def test_maptensor_put_get_export(ms_type): argument
70 def __init__(self, ms_type): argument
76 def construct(self, ms_type): argument
99 def test_mapparameter_ckpt_save_load(ms_type): argument
106 def __init__(self, ms_type): argument
112 def construct(self, ms_type): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/liteapi/include/ |
| D | datasets.h | 385 Status add_column(const std::string &name, mindspore::DataType ms_type) { in add_column()
394 Status add_column(const std::string &name, const std::string &ms_type) { in add_column()
403 …Status add_column(const std::string &name, mindspore::DataType ms_type, const std::vector<int32_t>… in add_column()
412 …Status add_column(const std::string &name, const std::string &ms_type, const std::vector<int32_t> … in add_column()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_aicpu_ops/ |
| D | test_gather_d.py | 60 def get_data(ms_type): argument
86 def test_net(ms_type): argument
105 def test_gatherd_dynamic(ms_type): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/gpu/ |
| D | test_ger_op.py | 43 def ger_graph(x1, x2, ms_type, nptype): argument
53 def ger_pynative(x1, x2, ms_type, nptype): argument
|
| D | test_selu_op.py | 22 from mindspore.common import dtype as ms_type unknown
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/minddata/dataset/include/dataset/ |
| D | datasets.h | 628 Status add_column(const std::string &name, mindspore::DataType ms_type) { in add_column()
637 Status add_column(const std::string &name, const std::string &ms_type) { in add_column()
646 …Status add_column(const std::string &name, mindspore::DataType ms_type, const std::vector<int32_t>… in add_column()
655 …Status add_column(const std::string &name, const std::string &ms_type, const std::vector<int32_t> … in add_column()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/test_tbe_ops/ |
| D | test_sqrt.py | 44 def sqrt_forward_test(ms_type, np_type): argument
57 def sqrt_backward_test(ms_type, np_type): argument
|
| D | test_assign_add.py | 40 def assign_add_forward_test(ms_type, np_type): argument
|
| D | test_minimum_grad.py | 49 def gen_data(inputA_np, inputB_np, grad_=None, ms_type=ms.float32): argument
|
| D | test_maximum_grad.py | 49 def gen_data(inputA_np, inputB_np, grad_=None, ms_type=ms.float32): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/aicpu/ |
| D | aicpu_util.cc | 63 int AicpuOpUtil::MsTypeToProtoType(TypeId ms_type) { in MsTypeToProtoType()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/ |
| D | common_utils.cc | 397 KernelAttr &KernelAttr::AddInputAttr(const TypeId &object_type, const TypeId &ms_type, const std::s… in AddInputAttr()
402 KernelAttr &KernelAttr::AddOptionalInputAttr(const TypeId &object_type, const TypeId &ms_type, in AddOptionalInputAttr()
408 KernelAttr &KernelAttr::AddOutputAttr(const TypeId &object_type, const TypeId &ms_type, const std::… in AddOutputAttr()
413 KernelAttr &KernelAttr::AddInputAttr(const TypeId &ms_type, const std::string &format) { in AddInputAttr()
418 KernelAttr &KernelAttr::AddOptionalInputAttr(const TypeId &ms_type, const std::string &format) { in AddOptionalInputAttr()
423 KernelAttr &KernelAttr::AddOutputAttr(const TypeId &ms_type, const std::string &format) { in AddOutputAttr()
458 void KernelAttr::SetInputAttr(const size_t index, const TypeId &ms_type, const std::string &format)… in SetInputAttr()
465 void KernelAttr::SetOutputAttr(const size_t index, const TypeId &ms_type, const std::string &format… in SetOutputAttr()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_exp_op.py | 19 def test_exp_tensor_api(ms_type): argument
|
| D | test_identity_op.py | 33 def generate_testcases(nptype, ms_type=None): argument
|
| D | test_identityn_op.py | 33 def generate_testcases(nptype, ms_type=None): argument
|
| D | test_stack_op.py | 33 def generate_testcases(nptype, ms_type=None): argument
|
| D | test_gather_elements_op.py | 33 def generate_testcases(ms_type=None): argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/dynamic_shape/ |
| D | test_lp_norm_op.py | 21 from mindspore.common import dtype as ms_type unknown
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/cpu/ |
| D | test_selu_op.py | 23 from mindspore.common import dtype as ms_type unknown
|
| D | test_mish_op.py | 23 from mindspore.common import dtype as ms_type unknown
|
| D | test_lrn_op.py | 23 from mindspore.common import dtype as ms_type unknown
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/sequence/ |
| D | sequence_addn_gpu_kernel.cc | 89 #define SEQUENCE_ADDN_KERNEL_REG(ms_type, builtin_type) \ argument
|
| D | sequence_stack_gpu_kernel.cc | 104 #define SEQUENCE_STACK_KERNEL_REG(ms_type, builtin_type) … argument
|
| /third_party/mindspore/mindspore-src/source/tests/st/pi_jit/share/ops/primitive/ |
| D | maximum_ops.py | 21 def __init__(self, left_input, right_input, dtype=np.float32, ms_type=None): argument
|
| D | minimum_ops.py | 20 def __init__(self, left_input, right_input, dtype=np.float32, grad=None, ms_type=None): argument
|