torch/_C/_itt.pyi

# Defined in torch/csrc/itt.cpp
def rangePush(message: str) -> None: ...
def rangePop() -> None: ...
def mark(message: str) -> None: ...