[pallas:fuser] Support full-dimension (dynamic_)slice and (dynamic_)update_slice
with non-multiple block sizes. Support (dynamic_)slice and (dynamic_)update_slice output fusion when tensor dimensions are not multiples of the block size, provided the slice/update spans the full dimension (i.e. full_size == slice_size). PiperOrigin-RevId: 972810803
R
Robert Dyro committed
05fcbe47999eb90b0a20714b28fcf3c834b09350
Parent: 264952c
Committed by jax authors <google-ml-automation@google.com>
on 8/28/2026, 10:08:10 PM