Searched defs:donated_invars (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 579 def _xla_call_impl(fun: lu.WrappedFun, *args, device, backend, name, donated_invars): argument
649 def _xla_callable(fun: lu.WrappedFun, device, backend, name, donated_invars, *arg_specs): argument
786 donated_invars=None): argument
897 call_jaxpr, donated_invars, device=None): argument
|
| H A D | pxla.py | 641 donated_invars, global_arg_shapes): argument
1222 donated_invars, global_arg_shapes): argument
|