| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 30 communication_hint='auto', argument
70 communication_hint=communication_hint.lower(),
80 communication_hint='auto', argument
118 communication_hint=communication_hint.lower(),
127 communication_hint='auto', argument
158 communication_hint=communication_hint.lower(),
166 communication_hint='auto', argument
198 communication_hint=communication_hint.lower(),
209 communication_hint='auto', argument
260 communication_hint=communication_hint.lower(),
[all …]
|
| D | collective_ops_gpu_test.py | 139 communication_hint='nccl'))
261 final_op='Id', communication_hint='nccl')
266 final_op='Id', communication_hint='nccl')
285 final_op='Id', communication_hint='NCCL')
307 communication_hint='nccl')
328 communication_hint='nccl')
339 communication_hint='nccl')
|
| D | collective_ops_test.py | 53 communication_hint='auto', argument
82 communication_hint=communication_hint,
180 communication_hint='nccl')
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 326 def _get_ordering_token(self, communication_hint): argument
327 if self._use_ordering_token() and communication_hint == 'NCCL':
338 communication_hint='AUTO', argument
354 ordering_token = self._get_ordering_token(communication_hint)
362 communication_hint=communication_hint,
366 def _all_gather(self, input_tensor, communication_hint='AUTO', timeout=0): argument
379 ordering_token = self._get_ordering_token(communication_hint)
386 communication_hint=communication_hint,
392 communication_hint='AUTO', argument
416 self.all_reduce(input_tensor, None, communication_hint, timeout))
[all …]
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 140 communication_hint=communication)
156 communication_hint=communication))
164 communication_hint=communication))
187 communication_hint=communication)
203 communication_hint=communication))
211 communication_hint=communication))
239 communication_hint=communication))
248 communication_hint=communication))
313 communication_hint=communication)
322 communication_hint=communication)
[all …]
|
| D | collective_ops_multi_worker_test.py | 120 communication_hint="ring")
191 communication_hint=communication)
209 communication_hint=communication)
219 communication_hint=communication)
231 communication_hint=communication)
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v2/ |
| D | CollectiveReduceV2.pbtxt | 59 name: "communication_hint"
125 name: "communication_hint"
203 name: "communication_hint"
|
| D | CollectiveGatherV2.pbtxt | 37 name: "communication_hint"
93 name: "communication_hint"
|
| D | CollectiveBcastSend.pbtxt | 83 name: "communication_hint"
132 name: "communication_hint"
|
| D | CollectiveBcastRecv.pbtxt | 75 name: "communication_hint"
120 name: "communication_hint"
|
| D | CollectiveGather.pbtxt | 82 name: "communication_hint"
130 name: "communication_hint"
|
| D | CollectiveReduce.pbtxt | 205 name: "communication_hint"
283 name: "communication_hint"
|
| D | CollectiveBcastSendV2.pbtxt | 38 name: "communication_hint"
|
| D | CollectiveBcastRecvV2.pbtxt | 51 name: "communication_hint"
|
| /external/tensorflow/tensorflow/core/ops/compat/ops_history_v1/ |
| D | CollectiveBcastRecv.pbtxt | 74 name: "communication_hint"
119 name: "communication_hint"
|
| D | CollectiveBcastSend.pbtxt | 82 name: "communication_hint"
131 name: "communication_hint"
|
| D | CollectiveGather.pbtxt | 82 name: "communication_hint"
|
| D | CollectiveReduce.pbtxt | 205 name: "communication_hint"
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | collective_ops.cc | 154 &col_params_->instance.impl_details.communication_hint)); in CollectiveGatherOpKernel()
242 &col_params_->instance.impl_details.communication_hint)); in CollectiveReduceOpKernel()
250 << col_params_->instance.impl_details.communication_hint in CollectiveReduceOpKernel()
336 &col_params_->instance.impl_details.communication_hint)); in CollectiveBcastSendOpKernel()
413 &col_params_->instance.impl_details.communication_hint)); in CollectiveBcastRecvOpKernel()
530 col_params->instance.impl_details.communication_hint = communication_hint_; in ComputeAsync()
669 col_params->instance.impl_details.communication_hint = communication_hint_; in ComputeAsync()
793 col_params->instance.impl_details.communication_hint = communication_hint_; in ComputeAsync()
927 col_params->instance.impl_details.communication_hint = communication_hint_; in ComputeAsync()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 98 string communication_hint; // user-supplied hint for implementation choice, member
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 677 (nccl_ || cp->instance.impl_details.communication_hint == "nccl") && in AssignCollectiveType()
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.raw_ops.pbtxt | 753 …T\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
757 …group_size\', \'group_key\', \'instance_key\', \'shape\', \'T\', \'communication_hint\', \'timeout…
761 …t\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
765 …rgs=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'communication_hint\', \'timeout…
769 …t\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
773 …oup_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'communication_hint\', \'timeout…
781 …', \'merge_op\', \'final_op\', \'subdiv_offsets\', \'wait_for\', \'communication_hint\', \'timeout…
785 …'instance_key\', \'ordering_token\', \'merge_op\', \'final_op\', \'communication_hint\', \'timeout…
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.raw_ops.pbtxt | 753 …T\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
757 …group_size\', \'group_key\', \'instance_key\', \'shape\', \'T\', \'communication_hint\', \'timeout…
761 …t\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
765 …rgs=[\'input\', \'group_size\', \'group_key\', \'instance_key\', \'communication_hint\', \'timeout…
769 …t\', \'group_size\', \'group_key\', \'instance_key\', \'shape\', \'communication_hint\', \'timeout…
773 …oup_size\', \'group_key\', \'instance_key\', \'ordering_token\', \'communication_hint\', \'timeout…
781 …', \'merge_op\', \'final_op\', \'subdiv_offsets\', \'wait_for\', \'communication_hint\', \'timeout…
785 …'instance_key\', \'ordering_token\', \'merge_op\', \'final_op\', \'communication_hint\', \'timeout…
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | ops.pbtxt | 7431 name: "communication_hint"
7496 name: "communication_hint"
7552 name: "communication_hint"
7604 name: "communication_hint"
7659 name: "communication_hint"
7715 name: "communication_hint"
7848 name: "communication_hint"
7926 name: "communication_hint"
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_generated_ops.td | 1838 DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
1859 DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
1882 DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
1908 DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
1933 DefaultValuedAttr<StrAttr, "auto">:$communication_hint,
|