Searched refs:backwards_graph (Results 1 – 1 of 1) sorted by relevance
622 def _create_forward_backward_with_graph(attrs, forward_graph, backwards_graph): argument637 backwards_graph, attrs=backward_function_attr)716 backwards_graph = func_graph_module.FuncGraph(719 name=backwards_graph.name,723 func_graph=backwards_graph)724 backwards_graph_captures = backwards_graph.external_captures737 self._attrs, self._func_graph, backwards_graph)915 backwards_graph = func_graph_module.FuncGraph(917 with backwards_graph.as_default():940 c for c in backwards_graph.external_captures[all …]