Searched refs:_gather_to_implementation (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/distribute/ |
| D | one_device_strategy.py | 400 def _gather_to_implementation(self, value, destinations, axis, options): member in OneDeviceExtended
|
| D | distribute_lib.py | 2456 return self._gather_to_implementation(value, destinations, axis, options)
2458 def _gather_to_implementation(self, value, destinations, axis, options): member in StrategyExtendedV2
3470 def _gather_to_implementation(self, value, destinations, axis, options): member in _DefaultDistributionExtended
|
| D | parameter_server_strategy.py | 504 def _gather_to_implementation(self, value, destinations, axis, member in ParameterServerStrategyExtended
|
| D | collective_all_reduce_strategy.py | 742 def _gather_to_implementation(self, value, destinations, axis, options): member in CollectiveAllReduceExtended
|
| D | mirrored_strategy.py | 709 def _gather_to_implementation(self, value, destinations, axis, options): member in MirroredExtended
|
| D | tpu_strategy.py | 1168 def _gather_to_implementation(self, value, destinations, axis, options): member in TPUExtended
|