| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 26 communication_hint='auto', argument
67 communication_hint=communication_hint.lower(),
101 communication_hint='auto', argument
152 communication_hint=communication_hint.lower(),
163 communication_hint='auto', argument
195 communication_hint=communication_hint.lower(),
203 communication_hint='auto', argument
242 communication_hint=communication_hint.lower(),
254 communication_hint='auto', argument
308 communication_hint=communication_hint.lower(),
[all …]
|
| D | collective_ops_gpu_test.py | 135 communication_hint='nccl'))
257 final_op='Id', communication_hint='nccl')
262 final_op='Id', communication_hint='nccl')
281 final_op='Id', communication_hint='NCCL')
303 communication_hint='nccl')
324 communication_hint='nccl')
335 communication_hint='nccl')
|
| D | collective_ops_test.py | 49 communication_hint='auto', argument
78 communication_hint=communication_hint,
176 communication_hint='nccl')
464 communication_hint='auto',
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | collective_ops_test.py | 149 communication_hint=communication,
167 communication_hint=communication))
176 communication_hint=communication))
205 communication_hint=communication)
222 communication_hint=communication))
231 communication_hint=communication))
254 communication_hint=communication))
286 communication_hint=communication))
295 communication_hint=communication))
391 communication_hint=communication)
[all …]
|
| D | collective_ops_multi_worker_test.py | 116 communication_hint="ring")
187 communication_hint=communication)
205 communication_hint=communication)
215 communication_hint=communication)
227 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"
289 name: "communication_hint"
383 name: "communication_hint"
477 name: "communication_hint"
|
| D | CollectiveGatherV2.pbtxt | 37 name: "communication_hint"
93 name: "communication_hint"
157 name: "communication_hint"
|
| D | CollectiveBcastRecv.pbtxt | 75 name: "communication_hint"
120 name: "communication_hint"
172 name: "communication_hint"
|
| D | CollectiveBcastSend.pbtxt | 83 name: "communication_hint"
132 name: "communication_hint"
188 name: "communication_hint"
|
| D | CollectiveGather.pbtxt | 82 name: "communication_hint"
130 name: "communication_hint"
185 name: "communication_hint"
|
| D | CollectiveBcastSendV2.pbtxt | 38 name: "communication_hint"
90 name: "communication_hint"
|
| D | CollectiveReduce.pbtxt | 205 name: "communication_hint"
283 name: "communication_hint"
369 name: "communication_hint"
455 name: "communication_hint"
|
| D | CollectiveBcastRecvV2.pbtxt | 51 name: "communication_hint"
116 name: "communication_hint"
|
| D | CollectiveInitializeCommunicator.pbtxt | 20 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 | 166 &col_params_->instance.impl_details.communication_hint)); in CollectiveGatherOpKernel()
258 &col_params_->instance.impl_details.communication_hint)); in CollectiveReduceOpKernel()
266 << col_params_->instance.impl_details.communication_hint in CollectiveReduceOpKernel()
352 &col_params_->instance.impl_details.communication_hint)); in CollectiveBcastSendOpKernel()
429 &col_params_->instance.impl_details.communication_hint)); in CollectiveBcastRecvOpKernel()
619 col_params->instance.impl_details.communication_hint = communication_hint_; in FillCollectiveParams()
911 string communication_hint, float timeout_seconds) in CollectiveGroupResource() argument
915 communication_hint_(communication_hint), in CollectiveGroupResource()
934 string communication_hint() const { return communication_hint_; } in communication_hint() function in tensorflow::__anon8b310e240111::CollectiveGroupResource
1090 col_params->instance.impl_details.communication_hint = in FillCollectiveParams()
[all …]
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2022-126.md | 20 communication_hint=arg_5, timeout_seconds=arg_6, name=arg_7)
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 359 communication_hint=options.implementation.value,
384 communication_hint=options.implementation.value,
|
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | xla_helpers.cc | 166 params->instance.impl_details.communication_hint = "nccl"; in ResolveDeviceAssignment()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 114 string communication_hint; // user-supplied hint for implementation choice, member
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | decompose_resource_ops.td | 618 $merge_op, $final_op, $communication_hint, $timeout_seconds,
623 $merge_op, $final_op, $communication_hint, $timeout_seconds,
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | collective_param_resolver_local.cc | 706 (nccl_ || cp->instance.impl_details.communication_hint == "nccl") && in AssignCollectiveType()
|