optimum/habana/transformers/models/vit/__init__.py (
1
lines of code) (
raw
):
from .modeling_vit import gaudi_vit_self_attention_forward