[Mosaic TPU] Support more convolution cases. #40781

Open
copybara-service[bot] opened 5:51pm on September 17, 2026 wants to merge 1 commit into google/jax main from
test_983521138

Pull Request #40781 commits