Searched refs:grad_graph (Results 1 – 1 of 1) sorted by relevance
399 def _resolve_grad_inputs(cond_graph, grad_graph): argument421 for t in grad_graph.external_captures:424 if t.graph != grad_graph.outer_graph:575 for grad_graph in grad_graphs:576 grad_graph.outputs = [577 t for t in func_graph_module.flatten(grad_graph.structured_outputs)