xla
Enable default buffer donation for gradient accumulation
#8758
Merged

Commits
  • Separate the buffer donor context manager
    rpsilva-aws committed 358 days ago
  • Introduce buffer donations to gradient accumulation API
    rpsilva-aws committed 358 days ago
  • Minor gradient accumulation refactoring
    rpsilva-aws committed 358 days ago
Loading