Searched refs:internal_captures (Results 1 – 13 of 13) sorted by relevance
298 internal_captures = []304 internal_captures = base_graph.internal_captures314 for s in internal_captures:
44 original_captures = concrete_fn.graph.internal_captures52 lifted_captures = g.internal_captures
166 existing_captures = {id(c) for c in graph.internal_captures}283 internal_captures = {id(c) for c in self.graph.internal_captures}284 flat_feeds = [f for f in flat_feeds if id(f) not in internal_captures]337 sources=flat_feeds + self.graph.internal_captures,
283 func_graph.inputs + func_graph.internal_captures)
954 gradients_wrt_outputs + backwards_graph.internal_captures)1158 + wrapped_backwards_graph.internal_captures)3080 graph_function.graph.internal_captures)
2732 internal_captures = g.internal_captures2733 self.assertLen(internal_captures, 1)2734 self.assertEqual(internal_captures[0].op.type, op_type)
39 name: "internal_captures"
753 def internal_captures(self): member in FuncGraph1030 inputs + func_graph.internal_captures + nest.flatten(
990 input_tensors = func.graph.internal_captures
1017 for i, capture in enumerate(tensor.op.graph.internal_captures):
210 body_graph.outputs.extend(body_graph.internal_captures)343 len(body_graph.inputs) - len(body_graph.internal_captures))
434 for i, output in enumerate(t.graph.internal_captures):
4137 exec_graph.inputs = exec_graph.internal_captures