# mypy: allow-untyped-defs from typing import TypeVar import torch T = TypeVar("T") # returns if all are the same mode def all_same_mode(modes): return all(tuple(mode == modes[0] for mode in modes)) no_dispatch = torch._C._DisableTorchDispatch