Searched refs:move_binders_to_front (Results 1 – 2 of 2) sorted by relevance
856 branch_jaxpr_2 = pe.move_binders_to_front(branch_jaxpr_2, move)1586 jaxpr_2_opt = pe.move_binders_to_front(jaxpr_2, move)
884 def move_binders_to_front(closed_jaxpr: ClosedJaxpr, to_move: Sequence[bool]) -> ClosedJaxpr: function