Searched refs:_reduce_shard_tensor (Results 1 – 2 of 2) sorted by relevance
183 def _reduce_shard_tensor( member in Shard613 return shard_spec._reduce_shard_tensor(tensor, mesh, self.reduce_op, mesh_dim)
150 return shard_spec._reduce_shard_tensor(tensor, mesh, self.reduce_op, mesh_dim)