| /third_party/mindspore/mindspore/ccsrc/backend/optimizer/ascend/mindir/ |
| D | all_to_all_unify_mindir.cc | 60 int64_t split_count = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateSplitNode() local
77 if (split_count == 0 || shape[LongToSize(split_dim)] % static_cast<size_t>(split_count) != 0) { in CreateSplitNode()
78 …MS_LOG(EXCEPTION) << "Invalid split count " << split_count << " cannot be divisible by shape[" << … in CreateSplitNode()
81 shape[LongToSize(split_dim)] /= static_cast<size_t>(split_count); in CreateSplitNode()
82 std::vector<TypeId> dtypes(split_count, dtype); in CreateSplitNode()
83 std::vector<std::vector<size_t>> shapes(split_count, shape); in CreateSplitNode()
86 AnfAlgo::SetNodeAttr(kAttrNumSplit, MakeValue<int64_t>(split_count), split_v); in CreateSplitNode()
87 …AnfAlgo::SetNodeAttr(kAttrSizeSplits, MakeValue(std::vector<int64_t>(split_count, shape[LongToSize… in CreateSplitNode()
97 int64_t split_count = AnfAlgo::GetNodeAttr<int64_t>(all_to_all, kAttrSplitCount); in CreateAllToAllvNode() local
100 CreateMultipleOutputsOfAnfNode(graph, split, static_cast<size_t>(split_count), &split_outputs); in CreateAllToAllvNode()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/tensor_layout/ |
| D | construct_operator.cc | 90 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in StridedSliceOP() local
91 if (split_count <= 0) { in StridedSliceOP()
122 if (num % split_count != 0) { in StridedSliceOP()
123 …MS_LOG(ERROR) << "Tensor can not be split into " << split_count << " slices in the dimension " << … in StridedSliceOP()
127 int64_t count = num / split_count; in StridedSliceOP()
178 Status ConstructOperator::SplitOP(int64_t split_count) { in SplitOP() argument
180 if (split_count <= 0) { in SplitOP()
187 ValuePtr attr_value_split = MakeValue(split_count); in SplitOP()
201 int64_t split_count = args[TRANSFER_PERMUTE_SPLIT_COUNT_INDEX]; in AlltoAllOP() local
205 if (split_count <= 0) { in AlltoAllOP()
[all …]
|
| D | redistribution_operator_infer.cc | 263 int64_t split_count = args[TRANSFER_CONCAT_SPLIT_COUNT_INDEX]; in TransferConcatByAxis() local
271 if (constructor_.SplitOP(split_count) != Status::SUCCESS) { in TransferConcatByAxis()
275 output_info_vector_.push_back(std::make_pair(true, split_count)); in TransferConcatByAxis()
|
| D | construct_operator.h | 42 Status SplitOP(int64_t split_count);
|
| /third_party/mindspore/tests/ut/cpp/parallel/tensor_layout/ |
| D | construct_operator_test.cc | 82 int64_t split_count = args[0]; in TEST_F() local
109 ASSERT_EQ(diff, num / split_count); in TEST_F()
125 int64_t split_count = 2; in TEST_F() local
126 ASSERT_EQ(constructor.SplitOP(split_count), Status::SUCCESS); in TEST_F()
130 int64_t split_count = 2; in TEST_F() local
135 Args args = {split_count, split_dim, concat_dim, dev_dim, dev_num}; in TEST_F()
|
| /third_party/mindspore/tests/ut/python/parallel/ |
| D | test_alltoall.py | 132 self.alltoallv = AlltoAll(split_count=8, split_dim=2, concat_dim=3)
153 self.alltoallv = AlltoAll(split_count=7, split_dim=2, concat_dim=3)
175 self.alltoallv = AlltoAll(split_count=[8], split_dim=2, concat_dim=3)
197 self.alltoallv = AlltoAll(split_count=8, split_dim=4, concat_dim=3)
219 self.alltoallv = AlltoAll(split_count=8, split_dim=(3,), concat_dim=3)
241 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=4)
263 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=([3],))
285 self.alltoallv = AlltoAll(split_count=3, split_dim=3, concat_dim=3)
307 self.alltoallv = AlltoAll(split_count=8, split_dim=3, concat_dim=3, group=3)
|
| /third_party/boost/libs/intrusive/test/ |
| D | unordered_test.hpp | 455 BOOST_TEST(testset1.split_count() == BucketSize/2); in test_rehash()
460 BOOST_TEST(testset1.split_count() == (BucketSize/2+1)); in test_rehash()
464 …for(std::size_t split_bucket = testset1.split_count(); split_bucket != BucketSize; ++split_bucket){ in test_rehash()
466 BOOST_TEST(testset1.split_count() == (split_bucket+1)); in test_rehash()
472 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
483 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
494 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
503 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
512 BOOST_TEST(testset1.split_count() == BucketSize); in test_rehash()
544 for(std::size_t split_bucket = testset1.split_count(); split_bucket > 6; --split_bucket){ in test_rehash()
[all …]
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | comm_ops.py | 681 def __init__(self, split_count, split_dim, concat_dim, group=GlobalComm.WORLD_COMM_GROUP): argument
684 validator.check_is_int(split_count, int)
687 self.split_count = split_count
695 if self.split_count != rank_size:
698 if x_shape[self.split_dim] % self.split_count != 0:
701 x_shape[self.concat_dim] = x_shape[self.concat_dim] * self.split_count
702 x_shape[self.split_dim] = int(x_shape[self.split_dim] / self.split_count)
|
| /third_party/mindspore/mindspore/lite/tools/optimizer/parallel/ |
| D | conv2d_info.cc | 41 int split_count = 0; in CheckStrategy() local
48 split_count++; in CheckStrategy()
54 split_count++; in CheckStrategy()
64 split_count++; in CheckStrategy()
70 split_count++; in CheckStrategy()
86 if (split_count > 1) { in CheckStrategy()
|
| D | depthwise_conv2d_info.cc | 131 int split_count = 0; in CheckStrategy() local
137 split_count++; in CheckStrategy()
144 split_count++; in CheckStrategy()
151 split_count++; in CheckStrategy()
158 split_count++; in CheckStrategy()
175 if (split_count > 1) { in CheckStrategy()
|
| /third_party/boost/libs/move/test/ |
| D | bench_merge.cpp | 61 std::size_t split_count = L / 2; in generate_elements() local
62 std::stable_sort(elements.data(), elements.data() + split_count, comp); in generate_elements()
63 std::stable_sort(elements.data() + split_count, elements.data() + L, comp); in generate_elements()
64 return split_count; in generate_elements()
|
| /third_party/mindspore/tests/ut/cpp/python_input/gtest_input/pre_activate/ |
| D | all_to_all_unify_mindir_test.py | 48 altoall = AlltoAll(split_count=2, split_dim=2, concat_dim=3)
|
| /third_party/boost/boost/intrusive/ |
| D | unordered_set.hpp | 373 size_type split_count() const;
854 size_type split_count() const;
|
| D | hashtable.hpp | 1469 BOOST_INTRUSIVE_FORCEINLINE SizeType split_count() const in split_count() function
1685 using internal_type::split_count;
3047 const size_type split_idx = this->split_count(); in incremental_rehash()
3087 size_type split_count() const;
|
| /third_party/mindspore/mindspore/ops/_grad/ |
| D | grad_comm_ops.py | 380 all_to_all_grad = AlltoAll(self.split_count, self.concat_dim, self.split_dim, self.group)
|
| /third_party/chromium/patch/ |
| D | 0003-3.2-Beta3-1115.patch | 252520 if (defined(invoker.split_count) && invoker.split_count > 1 &&
|