Searched refs:external_captures (Results 1 – 4 of 4) sorted by relevance
144 for t in cond_graph.external_captures:178 loop_vars = loop_vars + body_graph.external_captures186 num_cond_captures = len(cond_graph.external_captures)187 assert (cond_graph.external_captures ==188 body_graph.external_captures[:num_cond_captures])189 for body_capture in body_graph.external_captures[num_cond_captures:]:536 for t in body_grad_graph.external_captures:
84 true_graph.external_captures,85 false_graph.external_captures,339 for t in grad_graph.external_captures:351 t = t.graph.external_captures[index]
498 for capture in resource_initializer_function.graph.external_captures:
505 def external_captures(self): member in FuncGraph