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

Triggered via pull request September 29, 2024 07:37
@copybara-servicecopybara-service[bot]
opened #23994
test_680094115
Status Success
Total duration 9m 16s
Billable time 16m
Artifacts –

ci-build.yaml

on: pull_request
lint_and_typecheck
54s
lint_and_typecheck
FFI example
45s
FFI example
Matrix: build
Matrix: Documentation - test code snippets
Matrix: Documentation - render documentation
Matrix: jax2tf_test
Fit to window
Zoom out
Zoom in