Searched refs:backend_compile (Results 1 – 3 of 3) sorted by relevance
| /dports/math/py-jax/jax-0.2.9/jax/interpreters/ |
| H A D | xla.py | 288 compiled = backend_compile(backend, built_c, options)
349 def backend_compile(backend, built_c, options): function
727 compiled = backend_compile(backend, built, options)
1369 compiled = backend_compile(xb.get_device_backend(device), built_c, options)
|
| H A D | sharded_jit.py | 179 compiled = xla.backend_compile(
|
| H A D | pxla.py | 904 compiled = xla.backend_compile(backend, built, compile_options)
1511 compiled = xla.backend_compile(backend, built, compile_options)
1646 compiled = xla.backend_compile(backend, built, compile_options)
|