optimum/habana/transformers/models/albert/__init__.py (1 lines of code) (raw):

from .modeling_albert import gaudi_albert_forward