diff options
| author | David Riazati <davidriazati@fb.com> | 2018-12-18 11:43:45 -0800 |
|---|---|---|
| committer | Facebook Github Bot <facebook-github-bot@users.noreply.github.com> | 2018-12-18 12:03:08 -0800 |
| commit | 3118124cd6132588c4bf6614cbff1a3abb977bd9 (patch) | |
| tree | f0ad51b99c24b4729d3f59b0f5ba0a48926e766a /torch/jit | |
| parent | f4c504593cc73c4c3939f8d3e8e012c4d47bfd8d (diff) | |
| download | pytorch-3118124cd6132588c4bf6614cbff1a3abb977bd9.tar.gz pytorch-3118124cd6132588c4bf6614cbff1a3abb977bd9.tar.bz2 pytorch-3118124cd6132588c4bf6614cbff1a3abb977bd9.zip | |
Add (Un)Fold modules to standard library (#14759)
Summary:
Depends on #14597 for the corresponding aten ops.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14759
Differential Revision: D13325356
Pulled By: driazati
fbshipit-source-id: 99e39449c1ccfa293de05672c31a11e580bdd11f
Diffstat (limited to 'torch/jit')
| -rw-r--r-- | torch/jit/__init__.py | 3 |
1 files changed, 1 insertions, 2 deletions
diff --git a/torch/jit/__init__.py b/torch/jit/__init__.py index dc1499ffc8..ab06a34fac 100644 --- a/torch/jit/__init__.py +++ b/torch/jit/__init__.py @@ -1441,8 +1441,7 @@ def _get_builtin_table(): _builtin_table[id(torch.nn.functional.upsample_nearest)] = "aten::__upsample_nearest" _builtin_table[id(torch.nn.functional.upsample)] = "aten::__upsample" _builtin_table[id(torch.nn.functional.upsample_bilinear)] = "aten::__upsample_bilinear" - _builtin_table[id(torch.nn.functional.fold)] = "aten::fold" - _builtin_table[id(torch.nn.functional.unfold)] = "aten::unfold" + _builtin_table[id(torch.nn.functional.assert_int_or_pair)] = "aten::_assert_int_or_pair" return _builtin_table |