| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/ascend/optimizer/mindir/ |
| D | all_to_all_unify_mindir.cc | 62 const AnfNodePtr &input_node, int64_t split_count, in CreateSplitNode()
151 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateSplitNodeWithSplitDim() local
165 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateSplitNodeWithDim0() local
174 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateAllToAllvNode() local
223 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateAllToAllNode() local
247 const CNodePtr &input_node, int64_t split_count, in CreateConcatNode()
280 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateConcatNodeWithConcatDim() local
288 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateConcatNodeWithDim0() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/gpu/optimizer/ |
| D | alltoall_fusion.cc | 50 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateSplitNode() local
106 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateAllToAllvNode() local
144 int64_t split_count = common::AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateConcatNode() local
|
| /third_party/mindspore/mindspore-src/source/tests/ut/cpp/parallel/tensor_layout/ |
| D | construct_operator_test.cc | 82 int64_t split_count = args[0]; in TEST_F() local
125 int64_t split_count = 2; in TEST_F() local
130 int64_t split_count = 2; in TEST_F() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 140 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in ReplaceStridedSliceOpToSplitOp() local
182 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in StridedSliceOP() local
281 Status ConstructOperator::SplitOP(int64_t split_count) { in SplitOP()
304 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in AlltoAllOP() local
|
| D | redistribution_operator_infer.cc | 306 int64_t split_count = args[TRANSFER_CONCAT_SPLIT_COUNT_INDEX]; in TransferConcatByAxis() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/plugin/device/cpu/kernel/nnacl/infer/ |
| D | split_infer.c | 26 int split_count = 1; in UpdateSplitSize() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 44 int split_count = 0; in CheckStrategy() local
|
| D | depthwise_conv2d_info.cc | 139 int split_count = 0; in CheckStrategy() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ops_info/ |
| D | matmul_info.cc | 1230 size_t split_count = device_group_list[0].GetDevicesList().size(); in ComputePreAllGatherGraph() local
1397 int64_t split_count = dev_matrix_shape_[1]; in replace_graph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/frontend/parallel/ |
| D | step_assigned_parallel.cc | 887 int64_t split_count = devices; in FixReturnRedistribution() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/python/mindspore/ops/operations/ |
| D | comm_ops.py | 962 def __init__(self, split_count, split_dim, concat_dim, group=GlobalComm.WORLD_COMM_GROUP): argument
|