xla
22ba4af2 - fix amp dtype setting for GPU (#5337)

Commit

2 years ago

fix amp dtype setting for GPU (#5337) * fix amp dtype setting for GPU. * fix ut * fix lint. * minor.

References

Author

baoleai

Parents