optimum/habana/accelerate/state.py (1 lines of code) (raw):