HomeSort by relevance Sort by last modified time
    Searched refs:impl_details (Results 1 - 18 of 18) sorted by null

  /external/tensorflow/tensorflow/core/framework/
collective.cc 71 impl_details.subdiv_offsets.assign(
72 other.impl_details.subdiv_offsets.begin(),
73 other.impl_details.subdiv_offsets.end());
74 impl_details.subdiv_permutations.clear();
75 for (auto p : other.impl_details.subdiv_permutations) {
76 impl_details.subdiv_permutations.push_back(
79 impl_details.subdiv_source_rank.assign(
80 other.impl_details.subdiv_source_rank.begin(),
81 other.impl_details.subdiv_source_rank.end());
82 impl_details.dependencies = other.impl_details.dependencies
    [all...]
collective.h 98 CollImplDetails impl_details; member in struct:tensorflow::CollInstanceParams
  /external/tensorflow/tensorflow/core/common_runtime/
ring_alg.cc 118 col_params->instance.impl_details.collective_name);
137 col_params->instance.impl_details.collective_name);
142 col_params->instance.impl_details.subdiv_offsets.reserve(num_subdivs);
146 col_params->instance.impl_details.subdiv_offsets.push_back(subdiv_offset);
152 col_params->instance.impl_details.subdiv_offsets) {
193 if (col_params->instance.impl_details.subdiv_offsets.empty()) {
200 << &col_params->instance.impl_details.subdiv_permutations;
201 col_params->instance.impl_details.subdiv_permutations.resize(
202 col_params->instance.impl_details.subdiv_offsets.size());
204 col_params->instance.impl_details.subdiv_offsets.size(), -1)
    [all...]
ring_gatherer.cc 45 DCHECK_EQ(col_params->instance.impl_details.collective_name, "RingGather");
49 if (!col_params->instance.impl_details.subdiv_offsets.empty() &&
50 (col_params->instance.impl_details.subdiv_offsets.size() > 1 ||
51 col_params->instance.impl_details.subdiv_offsets[0] != 0)) {
55 if (col_params->instance.impl_details.subdiv_offsets.empty()) {
56 col_params->instance.impl_details.subdiv_offsets.push_back(0);
67 col_params_->instance.impl_details.subdiv_permutations.size());
77 sd < col_params_->instance.impl_details.subdiv_permutations.size();
81 col_params_->instance.impl_details.subdiv_permutations[sd]) {
hierarchical_tree_broadcaster.cc 79 CHECK_EQ(col_params->instance.impl_details.collective_name,
118 col_params->instance.impl_details.subdiv_permutations.resize(num_subdivs);
120 col_params->instance.impl_details.subdiv_source_rank.reserve(num_subdivs);
128 col_params->instance.impl_details.subdiv_permutations[sdi];
150 col_params->instance.impl_details.subdiv_source_rank.push_back(source_task);
161 col_params->instance.impl_details.subdiv_permutations[sdi];
175 col_params->instance.impl_details.subdiv_source_rank.push_back(
180 CHECK_GE(col_params->instance.impl_details.subdiv_source_rank[sri], 0);
223 const auto& impl = cp.instance.impl_details;
243 const auto& impl = cp.instance.impl_details;
    [all...]
collective_util.cc 56 col_params.instance.impl_details.subdiv_permutations;
68 for (auto o : col_params.instance.impl_details.subdiv_offsets)
74 for (auto src : col_params.instance.impl_details.subdiv_source_rank)
hierarchical_tree_broadcaster_test.cc 61 cp.instance.impl_details.subdiv_source_rank = {S}; \
62 cp.instance.impl_details.subdiv_permutations.push_back( \
263 col_params_.instance.impl_details.subdiv_offsets.clear();
268 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
274 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize(
278 col_params_.instance.impl_details
286 col_params_.instance.impl_details.subdiv_permutations[subdiv_i])
294 col_params_.instance.impl_details.subdiv_permutations[subdiv_i].resize(
299 << col_params_.instance.impl_details.subdiv_permutations.size();
303 col_params_.instance.impl_details.subdiv_permutations[subdiv_i][rank]
    [all...]
ring_reducer_test.cc 204 col_params_.instance.impl_details.subdiv_offsets.clear();
206 col_params_.instance.impl_details.collective_name = "RingReduce";
208 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
212 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi *
249 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) %
251 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back(
378 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets)
389 cp->instance.impl_details.subdiv_permutations.clear();
395 cp->instance.impl_details.subdiv_permutations);
425 col_params_.instance.impl_details.subdiv_permutations[0][rank]
    [all...]
ring_gatherer_test.cc 182 col_params_.instance.impl_details.subdiv_offsets.clear();
184 col_params_.instance.impl_details.collective_name = "RingGather";
186 col_params_.instance.impl_details.subdiv_permutations.resize(num_subdivs);
190 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi *
227 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) %
229 col_params_.instance.impl_details.subdiv_permutations[sdi].push_back(
366 cp->instance.impl_details.subdiv_permutations.clear();
372 cp->instance.impl_details.subdiv_permutations);
401 col_params_.instance.impl_details.subdiv_permutations[0][rank];
405 for (int r = 0; r < static_cast<int>(col_params_.instance.impl_details
    [all...]
ring_reducer.cc 49 CHECK_EQ(col_params->instance.impl_details.collective_name, "RingReduce");
59 col_params_->instance.impl_details.subdiv_permutations.size());
69 sd < col_params_->instance.impl_details.subdiv_permutations.size();
73 col_params_->instance.impl_details.subdiv_permutations[sd]) {
collective_param_resolver_local_test.cc 173 cp->instance.impl_details.subdiv_offsets.push_back(0);
196 EXPECT_EQ(cps[i].instance.impl_details.subdiv_source_rank.size(), 0);
216 cp->instance.impl_details.subdiv_offsets.push_back(0);
collective_param_resolver_local.cc 614 cp->instance.impl_details.collective_name = "HierarchicalTreeBroadcast";
617 cp->instance.impl_details.collective_name = "NcclReduce";
619 cp->instance.impl_details.collective_name = "RingReduce";
622 cp->instance.impl_details.collective_name = "RingGather";
624 cp->instance.impl_details.collective_name = "undef";
627 << cp->instance.impl_details.collective_name;
673 cp->instance.impl_details.collective_name, &col_impl);
688 cp->instance.impl_details.collective_name == "NcclReduce" &&
base_collective_executor.cc 300 col_params.instance.impl_details.collective_name, col_impl);
313 for (int32 instance : col_params.instance.impl_details.dependencies) {
  /external/tensorflow/tensorflow/core/kernels/
collective_ops.cc 54 col_params_.instance.impl_details.dependencies = dependencies_;
133 &col_params_.instance.impl_details.subdiv_offsets));
243 col_params_.instance.impl_details.subdiv_offsets = {0};
312 col_params_.instance.impl_details.subdiv_offsets = {0};
collective_nccl_reducer.cc 33 col_params->instance.impl_details.collective_name != "NcclReduce") {
37 col_params->instance.impl_details.collective_name,
collective_nccl_reducer_test.cc 125 col_params_.instance.impl_details.collective_name = "NcclReduce";
209 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets)
  /external/tensorflow/tensorflow/core/distributed_runtime/
collective_param_resolver_distributed.cc 62 for (int32 offset : instance.impl_details.subdiv_offsets) {
157 cp->instance.impl_details.subdiv_offsets.push_back(offset);
collective_param_resolver_distributed_test.cc 200 cp.instance.impl_details.subdiv_offsets.push_back(0);

Completed in 609 milliseconds