Searched defs:_gather_implementation (Results 1 – 1 of 1) sorted by relevance
/external/tensorflow/tensorflow/python/distribute/ |
D | cross_device_ops.py | 350 def _gather_implementation(self, per_replica_value, destinations, axis, member in CrossDeviceOps 603 def _gather_implementation(self, per_replica_value, destinations, axis, member in ReductionToOneDevice 917 def _gather_implementation(self, per_replica_value, destinations, axis, member in AllReduceCrossDeviceOps 1303 def _gather_implementation(self, per_replica_value, destinations, axis, member in CollectiveAllReduce
|