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

References

#35084 - [mosaic] Reverted `tpu_custom_call.lowered_as_tpu_kernel`

Author

superbobry

Committer

Google-ML-Automation

Parents

a18a7648

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

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