xla
3e7c301b - [POC] Add AMP zero gradients optimization (#3119)

Commit

4 years ago

[POC] Add AMP zero gradients optimization (#3119) * Add AMP zero gradients optimization * Address PR comment * Address PR comment

References

Author

ymwangg

Parents