Searched defs:xla_args (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 733 def set_up_aliases(c, xla_args, out_tuple, donated_args, tuple_args): argument
942 def f(c, *xla_args, **params):
946 def f_with_avals(c, avals, xla_args, params): argument
981 def f(c, axis_env, name_stack, avals, backend, *xla_args, **params):
|
| /dports/math/py-jax/jax-0.2.9/jax/ |
| H A D | api.py | 2552 def fun_translation(c, *xla_args, **params):
|