Searched defs:split_item (Results 1 – 8 of 8) sorted by relevance
| /third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/ |
| D | test_grouped_matmul.py | 61 …def __init__(self, np_w0, np_w1, split_item=3, group_type=0, mul_stra=None, gmm_stra=None, relu_st… argument
166 …def __init__(self, np_w0, np_b0, split_item=3, group_type=0, mul_stra=None, gmm_stra=None, relu_st… argument
311 …def __init__(self, np_w0, np_b0, split_item=3, group_type=0, mul_stra=None, gmm_stra=None, relu_st… argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/opapi/aclnn/ |
| D | grouped_matmul_aclnn_kernel.cc | 70 int64_t split_item = split_item_tensor->GetValueWithCheck<int64_t>(); in GetWorkSpaceInfo() local
116 int64_t split_item = split_item_tensor->GetValueWithCheck<int64_t>(); in Launch() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/kernel/pyboost/customize/ |
| D | grouped_matmul.cc | 33 … const std::optional<BaseTensorPtr> &group_list, const Int64ImmPtr &split_item, in GroupedMatmulGPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/pyboost/customize/ |
| D | grouped_matmul.cc | 33 … const std::optional<BaseTensorPtr> &group_list, const Int64ImmPtr &split_item, in GroupedMatmulCPUCustomize()
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/ |
| D | grouped_matmul.cc | 113 auto split_item = GetValue<int64_t>(split_ptr); in InferShape() local
227 auto split_item = GetValue<int64_t>(split_ptr); in InferType() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/kernel/pyboost/customize/ |
| D | grouped_matmul.cc | 42 … const std::optional<BaseTensorPtr> &group_list_tensor, const Int64ImmPtr &split_item, in GroupedMatmulAscendCustomize()
|
| /third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/ |
| D | test_grouped_matmul.py | 72 def __init__(self, split_item=3, group_type=-1): argument
|
| /third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/ |
| D | gen_lite_ops.cc | 2219 void GroupedMatmul::set_split_item(const int64_t &split_item) { (void)this->AddAttr("sp… in set_split_item()
|