| 123456789101112131415161718192021222324252627282930 |
- """
- This makes the functions in torch._C._VariableFunctions available as
- torch._VF.<funcname>
- without mypy being able to find them.
- A subset of those functions are mapped to ATen functions in
- torch/jit/_builtins.py
- See https://github.com/pytorch/pytorch/issues/21478 for the reason for
- introducing torch._VF
- """
- import sys
- import types
- import torch
- class VFModule(types.ModuleType):
- vf: types.ModuleType
- def __init__(self, name):
- super().__init__(name)
- self.vf = torch._C._VariableFunctions
- def __getattr__(self, attr):
- return getattr(self.vf, attr)
- sys.modules[__name__] = VFModule(__name__)
|
