Searched refs:grad_graphs (Results 1 – 1 of 1) sorted by relevance
554 def _create_zeros_for_none_grads(forward_graphs, grad_graphs): argument561 assert len(forward_graphs) == len(grad_graphs)562 branch_outputs = [g.structured_outputs for g in grad_graphs]570 with grad_graphs[branch_index].as_default():573 grad_graphs[branch_index].structured_outputs[output_idx] = zeros575 for grad_graph in grad_graphs: