jax

fc584cdf - Add tests for input dilation (LHS dilation / transposed convolution), kernel dilation (RHS dilation / atrous convolution), negative padding (cropping), and strided slicing in Mosaic TPU convolutions.

Add tests for input dilation (LHS dilation / transposed convolution), kernel dilation (RHS dilation / atrous convolution), negative padding (cropping), and strided slicing in Mosaic TPU convolutions. This is Part 2 of a 3-part changelist chain: - Part 2 (this CL): Dilations (input/LHS and kernel/RHS) test - Part 3 (follow-up CL): Grouped convolutions and property-based hypothesis testing. PiperOrigin-RevId: 969761722