Searched defs:_gather_to_implementation (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/distribute/ | ||
| H A D | collective_all_reduce_strategy.py | 800 def _gather_to_implementation(self, value, destinations, axis, options): member in CollectiveAllReduceExtended |
| H A D | one_device_strategy.py | 400 def _gather_to_implementation(self, value, destinations, axis, options): member in OneDeviceExtended |
| H A D | parameter_server_strategy.py | 507 def _gather_to_implementation(self, value, destinations, axis, member in ParameterServerStrategyExtended |
| H A D | mirrored_strategy.py | 720 def _gather_to_implementation(self, value, destinations, axis, options): member in MirroredExtended |
| H A D | distribute_lib.py | 2548 def _gather_to_implementation(self, value, destinations, axis, options): member in StrategyExtendedV2 |
| H A D | tpu_strategy.py | 1288 def _gather_to_implementation(self, value, destinations, axis, options): member in TPUExtended |