jax
6668ea68 - [mosaic] Reverted `tpu_custom_call.lowered_as_tpu_kernel`

Commit
98 days ago
[mosaic] Reverted `tpu_custom_call.lowered_as_tpu_kernel` It does have a few usages internally at Google. Reverts b38a8164f63a7e6827d79fef2e217b696c761f79 PiperOrigin-RevId: 869873558
Author
Parents
Loading