Adam Paszke 73275aa9ef [Mosaic] Add support for inserting a new lane dimension ...

This is often useful when a kernel uses statistics tensors that are constant across
the minormost dimensions. Right now the only way to use them is to force XLA to
insert the extra dimension before the kernel, but that turns out to be very inefficient.

PiperOrigin-RevId: 561903222

2023-09-01 03:02:02 -07:00

..

dialect/tpu

[Mosaic] Add support for inserting a new lane dimension

2023-09-01 03:02:02 -07:00

python

[Mosaic] Add support for inserting a new lane dimension

2023-09-01 03:02:02 -07:00

BUILD

[MOSAIC] apply_vector_layout C++ rewrite (1) VectorLayout functions

2023-08-29 22:57:13 -07:00