Searched refs:in_unknowns (Results 1 – 2 of 2) sorted by relevance
866 def _xla_call_partial_eval_update_params(params, in_unknowns): argument869 if not in_unknowns and donated_invars:874 donated_invars = [d for d, uk in zip(donated_invars, in_unknowns) if uk]
785 in_unknowns = ([False] * len(consts) +789 closed_jaxpr, in_unknowns, instantiate=False) # type: ignore792 … closed_jaxpr, in_unknowns, instantiate=False, trace_type=trace.main.trace_type) # type: ignore