Searched refs:set_up_aliases (Results 1 – 3 of 3) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 714 donated_invars = set_up_aliases(c, xla_args, out_tuple, donated_invars, tuple_args)
733 def set_up_aliases(c, xla_args, out_tuple, donated_args, tuple_args): function
|
| H A D | pxla.py | 850 donated_invars = xla.set_up_aliases(c, xla_args, out_tuple, donated_invars, tuple_args)
1493 donated_invars = xla.set_up_aliases(c, xla_args, out_tuple, donated_invars, tuple_args)
|
| /dports/math/py-jax/jax-0.2.9/jax/ |
| H A D | api.py | 673 donated_invars = xla.set_up_aliases(c, xla_args, out_tuple, donated_invars,
|