| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | ring_gatherer.cc | 51 if (!col_params->instance.impl_details.subdiv_offsets.empty() && in InitializeCollectiveParams()
52 (col_params->instance.impl_details.subdiv_offsets.size() > 1 || in InitializeCollectiveParams()
53 col_params->instance.impl_details.subdiv_offsets[0] != 0)) { in InitializeCollectiveParams()
57 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
58 col_params->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParams()
|
| D | ring_alg.cc | 143 col_params->instance.impl_details.subdiv_offsets.reserve(num_subdivs); in GenerateSubdivsInCollectiveParams()
147 col_params->instance.impl_details.subdiv_offsets.push_back(subdiv_offset); in GenerateSubdivsInCollectiveParams()
153 col_params->instance.impl_details.subdiv_offsets) { in GenerateSubdivsInCollectiveParams()
194 if (col_params->instance.impl_details.subdiv_offsets.empty()) { in InitializeCollectiveParams()
203 col_params->instance.impl_details.subdiv_offsets.size()); in InitializeCollectiveParams()
205 col_params->instance.impl_details.subdiv_offsets.size(), -1); in InitializeCollectiveParams()
207 sdi < col_params->instance.impl_details.subdiv_offsets.size(); ++sdi) { in InitializeCollectiveParams()
211 int offset = col_params->instance.impl_details.subdiv_offsets[sdi]; in InitializeCollectiveParams()
|
| D | ring_reducer_test.cc | 216 col_params_->instance.impl_details.subdiv_offsets.clear(); in Init()
224 col_params_->instance.impl_details.subdiv_offsets.push_back( in Init()
262 (di + col_params_->instance.impl_details.subdiv_offsets[sdi]) % in Init()
389 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduce()
584 cp->instance.impl_details.subdiv_offsets.push_back(0); in SetUpCollectiveParams()
604 cp->instance.impl_details.subdiv_offsets = {0, 4}; in TEST_F()
612 cp->instance.impl_details.subdiv_offsets = {0, -4}; in TEST_F()
621 cp->instance.impl_details.subdiv_offsets = {3, -3}; in TEST_F()
639 cp->instance.impl_details.subdiv_offsets.clear(); in TEST_F()
654 cp->instance.impl_details.subdiv_offsets.clear(); in TEST_F()
[all …]
|
| D | ring_gatherer_test.cc | 193 col_params_->instance.impl_details.subdiv_offsets.clear(); in Init()
201 col_params_->instance.impl_details.subdiv_offsets.push_back( in Init()
238 (di + col_params_->instance.impl_details.subdiv_offsets[sdi]) % in Init()
552 cp->instance.impl_details.subdiv_offsets.push_back(0); in SetUpCollectiveParams()
572 cp->instance.impl_details.subdiv_offsets = {}; in TEST_F()
577 cp->instance.impl_details.subdiv_offsets = {0}; in TEST_F()
583 cp->instance.impl_details.subdiv_offsets = {}; in TEST_F()
|
| D | collective_util.cc | 68 for (auto o : col_params.instance.impl_details.subdiv_offsets) in SubdivPermDebugString()
|
| D | collective_param_resolver_local_test.cc | 178 cp->instance.impl_details.subdiv_offsets.push_back(0); in TEST_F()
222 cp->instance.impl_details.subdiv_offsets.push_back(0); in InitializeCollectiveParamsForBroadcast()
|
| D | hierarchical_tree_broadcaster_test.cc | 277 col_params_->instance.impl_details.subdiv_offsets.clear(); in Init()
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveReduce.pbtxt | 59 name: "subdiv_offsets"
122 name: "subdiv_offsets"
193 name: "subdiv_offsets"
271 name: "subdiv_offsets"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 29 subdiv_offsets=(0,), argument
69 subdiv_offsets=subdiv_offsets,
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveReduce.pbtxt | 59 name: "subdiv_offsets"
122 name: "subdiv_offsets"
193 name: "subdiv_offsets"
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.cc | 83 impl_details.subdiv_offsets.assign( in operator =()
84 other.impl_details.subdiv_offsets.begin(), in operator =()
85 other.impl_details.subdiv_offsets.end()); in operator =()
109 for (const auto& d : impl_details.subdiv_offsets) { in ToString()
|
| D | collective.h | 94 std::vector<int> subdiv_offsets; member
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 224 &col_params_->instance.impl_details.subdiv_offsets)); in CollectiveReduceOpKernel()
341 col_params_->instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastSendOpKernel()
418 col_params_->instance.impl_details.subdiv_offsets = {0}; in CollectiveBcastRecvOpKernel()
534 col_params->instance.impl_details.subdiv_offsets.push_back(0); in ComputeAsync()
798 col_params->instance.impl_details.subdiv_offsets.push_back(0); in ComputeAsync()
932 col_params->instance.impl_details.subdiv_offsets.push_back(0); in ComputeAsync()
|
| D | collective_nccl_test.cc | 241 .Attr("subdiv_offsets", params.instance.impl_details.subdiv_offsets) in GetCollectiveReduceOpKernel()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | collective_param_resolver_distributed.cc | 72 for (int32 offset : instance.impl_details.subdiv_offsets) { in CompleteInstanceCall()
196 cp->instance.impl_details.subdiv_offsets.push_back(offset); in CompleteInstanceAsync()
|
| D | collective_param_resolver_distributed_test.cc | 204 cp->instance.impl_details.subdiv_offsets.push_back(0); in CreateCollectiveParams()
|
| /external/tensorflow/tensorflow/python/ops/numpy_ops/g3doc/ |
| D | TensorFlow_Numpy_Distributed_Image_Classification.ipynb | 468 " subdiv_offsets=[])))\n",
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 781 …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 | 781 …ze\', \'group_key\', \'instance_key\', \'merge_op\', \'final_op\', \'subdiv_offsets\', \'wait_for\…
|
| /external/tensorflow/tensorflow/go/op/ |
| D | wrappers.go | 43228 …oup_key int64, instance_key int64, merge_op string, final_op string, subdiv_offsets []int64, optio…
43232 …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"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1906 I64ArrayAttr:$subdiv_offsets,
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7836 name: "subdiv_offsets"
|