Home
last modified time | relevance | path

Searched defs:split_item (Results 1 – 8 of 8) sorted by relevance

/third_party/mindspore/mindspore-src/source/tests/ut/python/parallel/
Dtest_grouped_matmul.py61 …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/
Dgrouped_matmul_aclnn_kernel.cc70 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/
Dgrouped_matmul.cc33 … 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/
Dgrouped_matmul.cc33 … const std::optional<BaseTensorPtr> &group_list, const Int64ImmPtr &split_item, in GroupedMatmulCPUCustomize()
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/ops_func_impl/
Dgrouped_matmul.cc113 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/
Dgrouped_matmul.cc42 … const std::optional<BaseTensorPtr> &group_list_tensor, const Int64ImmPtr &split_item, in GroupedMatmulAscendCustomize()
/third_party/mindspore/mindspore-src/source/tests/st/ops/ascend/
Dtest_grouped_matmul.py72 def __init__(self, split_item=3, group_type=-1): argument
/third_party/mindspore/mindspore-src/source/mindspore/core/ops/auto_generate/
Dgen_lite_ops.cc2219 void GroupedMatmul::set_split_item(const int64_t &split_item) { (void)this->AddAttr("sp… in set_split_item()