| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer.cc | 49 if (!col_params->instance.impl_details.subdiv_offsets.empty() && in InitializeCollectiveParams()
50 (col_params->instance.impl_details.subdiv_offsets.size() > 1 || in InitializeCollectiveParams()
51 col_params->instance.impl_details.subdiv_offsets[0] != 0)) { in InitializeCollectiveParams()
55 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
56 col_params->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParams()
|
| D | ring_alg.cc | 142 col_params->instance.impl_details.subdiv_offsets.reserve(num_subdivs); in GenerateSubdivsInCollectiveParams()
146 col_params->instance.impl_details.subdiv_offsets.push_back(subdiv_offset); in GenerateSubdivsInCollectiveParams()
152 col_params->instance.impl_details.subdiv_offsets) { in GenerateSubdivsInCollectiveParams()
193 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
202 col_params->instance.impl_details.subdiv_offsets.size()); in InitializeCollectiveParams()
204 col_params->instance.impl_details.subdiv_offsets.size(), -1); in InitializeCollectiveParams()
206 sdi < col_params->instance.impl_details.subdiv_offsets.size(); ++sdi) { in InitializeCollectiveParams()
210 int offset = col_params->instance.impl_details.subdiv_offsets[sdi]; in InitializeCollectiveParams()
|
| D | ring_reducer_test.cc | 204 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
212 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi * in Init()
249 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) % in Init()
378 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduce()
569 cp.instance.impl_details.subdiv_offsets.push_back(0); in SetUpCollectiveParams()
588 cp.instance.impl_details.subdiv_offsets = {0, 4}; in TEST_F()
596 cp.instance.impl_details.subdiv_offsets = {0, -4}; in TEST_F()
605 cp.instance.impl_details.subdiv_offsets = {3, -3}; in TEST_F()
622 cp.instance.impl_details.subdiv_offsets.clear(); in TEST_F()
637 cp.instance.impl_details.subdiv_offsets.clear(); in TEST_F()
[all …]
|
| D | ring_gatherer_test.cc | 182 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
190 col_params_.instance.impl_details.subdiv_offsets.push_back(sdi * in Init()
227 (di + col_params_.instance.impl_details.subdiv_offsets[sdi]) % in Init()
542 cp.instance.impl_details.subdiv_offsets.push_back(0); in SetUpCollectiveParams()
561 cp.instance.impl_details.subdiv_offsets = {}; in TEST_F()
566 cp.instance.impl_details.subdiv_offsets = {0}; in TEST_F()
572 cp.instance.impl_details.subdiv_offsets = {}; in TEST_F()
|
| D | collective_param_resolver_local_test.cc | 173 cp->instance.impl_details.subdiv_offsets.push_back(0); in TEST_F()
216 cp->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParamsForBroadcast()
|
| D | collective_util.cc | 68 for (auto o : col_params.instance.impl_details.subdiv_offsets) in SubdivPermDebugString()
|
| D | hierarchical_tree_broadcaster_test.cc | 263 col_params_.instance.impl_details.subdiv_offsets.clear(); in Init()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 25 subdiv_offsets=(0,)): argument
58 subdiv_offsets=subdiv_offsets)
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 71 impl_details.subdiv_offsets.assign( in operator =()
72 other.impl_details.subdiv_offsets.begin(), in operator =()
73 other.impl_details.subdiv_offsets.end()); in operator =()
106 for (const auto& d : impl_details.subdiv_offsets) { in ToString()
|
| D | collective.h | 72 std::vector<int> subdiv_offsets; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 133 &col_params_.instance.impl_details.subdiv_offsets)); in CollectiveReduceOpKernel()
243 col_params_.instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastSendOpKernel()
312 col_params_.instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastRecvOpKernel()
|
| D | collective_nccl_reducer_test.cc | 209 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduce()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 62 for (int32 offset : instance.impl_details.subdiv_offsets) { in CompleteInstanceCall()
157 cp->instance.impl_details.subdiv_offsets.push_back(offset); in CompleteInstanceAsync()
|
| D | collective_param_resolver_distributed_test.cc | 200 cp.instance.impl_details.subdiv_offsets.push_back(0); in DefineCollectiveParams()
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 358 subdiv_offsets = [0] # TODO(tucker): maybe support non-default subdiv spec
368 unary_op, subdiv_offsets)
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 593 …ze\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offsets\', \'wait_for\…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 593 …ze\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offsets\', \'wait_for\…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 4121 …oup_key int64, instance_key int64, merge_op string, final_op string, subdiv_offsets []int64, optio…
4125 …e_key": instance_key, "merge_op": merge_op, "final_op": final_op, "subdiv_offsets": subdiv_offsets}
|
| /external/tensorflow/tensorflow/core/ops/compat/ |
| D | ops_history.v2.pbtxt | 13101 name: "subdiv_offsets"
13164 name: "subdiv_offsets"
|
| D | ops_history.v1.pbtxt | 13159 name: "subdiv_offsets"
13222 name: "subdiv_offsets"
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 5690 name: "subdiv_offsets"
|