Searched refs:_dce_open_jaxpr (Results 1 – 1 of 1) sorted by relevance
206 jaxpr = _dce_open_jaxpr(jaxpr, out_unknowns, drop_outputs=True)843 new_jaxpr = _dce_open_jaxpr(closed_jaxpr.jaxpr, tuple(outputs), drop_outputs)847 def _dce_open_jaxpr(jaxpr: Jaxpr, outputs: Tuple[bool, ...], drop_outputs=False) -> Jaxpr: function