SIGN IN SIGN UP

[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