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