Searched refs:add_capture (Results 1 – 4 of 4) sorted by relevance
132 graph.add_capture(new_variable.handle, old_variable.handle)348 pruned_graph.add_capture(external_capture, lift_map[internal_capture])
1022 forward_wrapper_graph.add_capture(capture, input_placeholder)1045 forward_wrapper_graph.add_capture(external_jvp, jvp_placeholder)
663 self.add_capture(tensor, placeholder)676 def add_capture(self, tensor, placeholder): member in FuncGraph735 self.add_capture(tensor, graph_const)
87 name: "add_capture"