You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
16 lines
255 B
16 lines
255 B
4 days ago
|
# 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
|