Searched defs:_gather (Results 1 – 3 of 3) sorted by relevance
54 def _gather(strategy, value): function
303 def _gather(self, per_replica_value, destinations, axis, options=None): member in CrossDeviceOps
114 def _gather(params, indices, axis, batch_dims): function