pytorch

1
# mypy: allow-untyped-defs
2
import torch
3
from typing import TypeVar
4

5
T = TypeVar('T')
6

7
# returns if all are the same mode
8
def all_same_mode(modes):
9
    return all(tuple(mode == modes[0] for mode in modes))
10

11
no_dispatch = torch._C._DisableTorchDispatch
12