Searched defs:communication_hint (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ |
| D | collective_ops.py | 30 communication_hint='auto', argument
80 communication_hint='auto', argument
127 communication_hint='auto', argument
166 communication_hint='auto', argument
209 communication_hint='auto', argument
268 communication_hint='auto', argument
303 communication_hint='auto', argument
346 communication_hint='auto', argument
|
| D | collective_ops_test.py | 53 communication_hint='auto', argument
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | cross_device_utils.py | 326 def _get_ordering_token(self, communication_hint): argument
338 communication_hint='AUTO', argument
366 def _all_gather(self, input_tensor, communication_hint='AUTO', timeout=0): argument
392 communication_hint='AUTO', argument
440 communication_hint='AUTO', argument
506 communication_hint='AUTO', argument
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | collective.h | 98 string communication_hint; // user-supplied hint for implementation choice, member
|