Searched refs:summed_grads (Results 1 – 2 of 2) sorted by relevance
391 summed_grads = nccl_ops.all_sum(scaled_grads)393 summed_grads = all_reduce.build_ring_all_reduce(396 summed_grads = all_reduce.build_nccl_then_ring(scaled_grads, num_shards,399 summed_grads = all_reduce.build_nccl_then_recursive_hd(402 summed_grads = all_reduce.build_nccl_then_shuffle(406 summed_grads = all_reduce.build_shuffle_then_shuffle(409 summed_grads = all_reduce.build_shuffle_all_reduce(415 for (_, v), g in zip(grad_and_vars, summed_grads):
92 Status ProcessWhileLoop(Node* exit_node, const Output& summed_grads);391 const Output& summed_grads) { in ProcessWhileLoop() argument396 if (summed_grads == NoGradient()) { in ProcessWhileLoop()408 backprops[exit_node] = summed_grads; in ProcessWhileLoop()