Searched refs:dy_s (Results 1 – 1 of 1) sorted by relevance
3132 return ys, lambda *dy_s: self.all_reduce(reduce_op, dy_s)3297 def grad(*dy_s): argument3298 grads = self.all_reduce(reduce_util.ReduceOp.SUM, dy_s)