xla
2c2f554a - Add optimization barrier to enable the gradient checkpointing (#3482)

Commit

3 years ago

Add optimization barrier to enable the gradient checkpointing (#3482) * Add optimization barrier * Make optimization_barrier take vector and add test for correctness

References

#3482 - Add optimization barrier to enable the gradient checkpointing

Author

JackCaoG

Parents

5d9c53d7

xla 2c2f554a - Add optimization barrier to enable the gradient checkpointing (#3482)

xla
2c2f554a - Add optimization barrier to enable the gradient checkpointing (#3482)