mirrors/rocm_jax
1
0
Fork 0
mirror of https://github.com/ROCm/jax.git synced 2025-04-16 11:56:07 +00:00
Code Issues Packages Projects Releases Wiki Activity
rocm_jax/docs/sphinxext
History
George Necula d323ad0f2b [host_callback] Add support for tapping empty arrays 
We make sure that both the inputs and the outputs of
callbacks can contain empty arrays.
Most platforms do not support empty infeed, so we ensure
we do not send those.
2021-03-30 10:48:58 +03:00
..
jax_extensions.py
[host_callback] Add support for tapping empty arrays
2021-03-30 10:48:58 +03:00