xla
b214c104 - Allow running with bfloat16 on XLA:GPU autocast (#5598)

Commit

2 years ago

Allow running with bfloat16 on XLA:GPU autocast (#5598) * Enable xla autocast for XLA:GPU * XLA autocast test for GPU and TPU

References

Author

yeounoh

Parents