mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-06 12:20:52 +01:00
See #127836 for details. Pull Request resolved: https://github.com/pytorch/pytorch/pull/127838 Approved by: https://github.com/oulgen
43 lines
654 B
Python
43 lines
654 B
Python
# ${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}
|