Searched refs:_drop_invars (Results 1 – 1 of 1) sorted by relevance
205 jaxpr = _drop_invars(jaxpr, in_knowns)868 def _drop_invars(jaxpr: Jaxpr, drop: Tuple[bool, ...]): function