| # ${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} |