Home
last modified time | relevance | path

Searched defs:_gather_implementation (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/python/distribute/
Dcross_device_ops.py350 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