Skip to content
CI

[Mosaic] Extend tpu matmulop to have dimension dims, modeled on jax's dot_general. Add support for batch dims via canonicalization to slice->matmul->concat #62625

Sign in to view logs

[Mosaic] Extend tpu matmulop to have dimension dims, modeled on jax's dot_general. Add support for batch dims via canonicalization to slice->matmul->concat

[Mosaic] Extend tpu matmulop to have dimension dims, modeled on jax's dot_general. Add support for batch dims via canonicalization to slice->matmul->concat #62625

FFI example

FFI example
succeeded Sep 29, 2024 in 45s