| # ${generated_comment} | |
| # mypy: disable-error-code="type-arg" | |
| # mypy: allow-untyped-defs | |
| import builtins | |
| from typing import ( | |
| Any, | |
| Callable, | |
| ContextManager, | |
| Iterator, | |
| List, | |
| Literal, | |
| NamedTuple, | |
| Optional, | |
| overload, | |
| Sequence, | |
| Tuple, | |
| TypeVar, | |
| Union, | |
| ) | |
| import torch | |
| from torch import contiguous_format, Generator, inf, memory_format, strided, SymInt, Tensor | |
| from torch.types import ( | |
| _bool, | |
| _complex, | |
| _device, | |
| _dtype, | |
| _float, | |
| _int, | |
| _layout, | |
| _qscheme, | |
| _size, | |
| Device, | |
| Number, | |
| ) | |
| from torch._prims_common import DeviceLikeType | |
| ${function_hints} | |
| ${all_directive} |