Searched refs:internal_captures (Results 1 – 6 of 6) sorted by relevance
117 existing_captures = set(self.graph.internal_captures)156 internal_captures = self.graph.internal_captures158 if f not in internal_captures]185 [sink_tensor], pruned_graph, sources=flat_feeds + internal_captures)
161 func_graph.inputs + func_graph.internal_captures)
1343 captured = frozenset(graph_function.graph.internal_captures)
181 body_graph.outputs.extend(body_graph.internal_captures)460 for internal_capture in grad_func_graph.internal_captures:
350 index = t.graph.internal_captures.index(t)
510 def internal_captures(self): member in FuncGraph