Searched refs:subdiv_source_rank (Results 1 – 6 of 6) sorted by relevance
120 col_params->instance.impl_details.subdiv_source_rank.reserve(num_subdivs); in InitializeCollectiveParams()150 col_params->instance.impl_details.subdiv_source_rank.push_back(source_task); in InitializeCollectiveParams()175 col_params->instance.impl_details.subdiv_source_rank.push_back( in InitializeCollectiveParams()180 CHECK_GE(col_params->instance.impl_details.subdiv_source_rank[sri], 0); in InitializeCollectiveParams()224 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size())); in TreeRecvFrom()225 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeRecvFrom()244 DCHECK_LT(subdiv, static_cast<int>(impl.subdiv_source_rank.size())); in TreeSendTo()245 int source_rank = impl.subdiv_source_rank[subdiv]; in TreeSendTo()301 int source_rank = col_params_->instance.impl_details.subdiv_source_rank[si]; in RunTree()
61 cp.instance.impl_details.subdiv_source_rank = {S}; \431 [0][cp.instance.impl_details.subdiv_source_rank[0]]; in RunTest()523 cp->instance.impl_details.subdiv_source_rank.clear(); in RunSubdivPermsTest()531 cp->instance.impl_details.subdiv_source_rank); in RunSubdivPermsTest()584 impl.subdiv_source_rank.resize(num_subdivs, 0); in DeviceInstance()603 col_params_.subdiv_rank[0] == impl.subdiv_source_rank[0]; in DeviceInstance()
74 for (auto src : col_params.instance.impl_details.subdiv_source_rank) in SubdivPermDebugString()
196 EXPECT_EQ(cps[i].instance.impl_details.subdiv_source_rank.size(), 0); in TEST_F()
79 impl_details.subdiv_source_rank.assign( in operator =()80 other.impl_details.subdiv_source_rank.begin(), in operator =()81 other.impl_details.subdiv_source_rank.end()); in operator =()117 if (!impl_details.subdiv_source_rank.empty()) { in ToString()119 for (const auto& r : impl_details.subdiv_source_rank) { in ToString()
73 std::vector<int> subdiv_source_rank; // rank of source in each subdiv member