mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
Summary: Context: https://github.com/pytorch/pytorch/pull/53299#discussion_r587882857 These are the only hand-written parts of this diff: - the addition to `.github/workflows/lint.yml` - the file endings changed in these four files (to appease FB-internal land-blocking lints): - `GLOSSARY.md` - `aten/src/ATen/core/op_registration/README.md` - `scripts/README.md` - `torch/csrc/jit/codegen/fuser/README.md` The rest was generated by running this command (on macOS): ``` git grep -I -l ' $' -- . ':(exclude)**/contrib/**' ':(exclude)third_party' | xargs gsed -i 's/ *$//' ``` I looked over the auto-generated changes and didn't see anything that looked problematic. Pull Request resolved: https://github.com/pytorch/pytorch/pull/53406 Test Plan: This run (after adding the lint but before removing existing trailing spaces) failed: - https://github.com/pytorch/pytorch/runs/2043032377 This run (on the tip of this PR) succeeded: - https://github.com/pytorch/pytorch/runs/2043296348 Reviewed By: walterddr, seemethere Differential Revision: D26856620 Pulled By: samestep fbshipit-source-id: 3f0de7f7c2e4b0f1c089eac9b5085a58dd7e0d97
24 lines
749 B
Python
24 lines
749 B
Python
import torch
|
|
from .. import Tensor
|
|
from typing import Tuple, Optional
|
|
import builtins
|
|
|
|
class Parameter(Tensor):
|
|
def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ...
|
|
|
|
...
|
|
|
|
def is_lazy(param: Tensor): ...
|
|
|
|
class UninitializedParameter(Tensor):
|
|
def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ...
|
|
|
|
def materialize(self, shape: Tuple[int, ...], device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None): ...
|
|
...
|
|
|
|
class UninitializedBuffer(Tensor):
|
|
def __init__(self, data: Tensor=..., requires_grad: builtins.bool=...): ...
|
|
|
|
def materialize(self, shape: Tuple[int, ...], device: Optional[torch.device] = None, dtype: Optional[torch.dtype] = None): ...
|
|
...
|