xla
d9523716 - Reduce unnecessary tensor allocation in Adam and AdamW (#5700)

Commit

2 years ago

Reduce unnecessary tensor allocation in Adam and AdamW (#5700) * Reduce unnecessary tensor allocation in adam and adamw * fix lint

References

Author

baoleai

Parents