https://github.com/google/jax
History
Tip revision: 8fb4e08f57e063c9f5ec2cc3ee1ebc405fbd5d01 authored by Jevin Jiang on 28 March 2024, 05:11:38 UTC
[XLA:Mosaic] Add type-convert-insertion pass and support shapecast a vector whose dtype is packed and sublane dim size is 1.
Tip revision: 8fb4e08
File Mode Size
__init__.py -rw-r--r-- 870 bytes

back to top