Searched refs:_execute_trivial (Results 1 – 1 of 1) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 679 return partial(_execute_trivial, jaxpr, device, consts, out_avals, result_handlers)
853 def _execute_trivial(jaxpr, device: Optional[Device], consts, avals, handlers, *args): function
|