Viewing File: /home/ubuntu/combine_ai/combine/lib/python3.10/site-packages/torch/_dynamo/variables/__init__.py
from .base import VariableTracker
from .builtin import BuiltinVariable
from .constant import ConstantVariable, EnumVariable
from .ctx_manager import (
ContextWrappingVariable,
DeterministicAlgorithmsVariable,
DisabledSavedTensorsHooksVariable,
GradModeVariable,
InferenceModeVariable,
StreamContextVariable,
StreamVariable,
WithExitFunctionVariable,
)
from .dicts import (
ConstDictVariable,
CustomizedDictVariable,
DataClassVariable,
DefaultDictVariable,
SetVariable,
)
from .functions import (
NestedUserFunctionVariable,
UserFunctionVariable,
UserMethodVariable,
)
from .higher_order_ops import TorchHigherOrderOperatorVariable
from .iter import (
CountIteratorVariable,
CycleIteratorVariable,
IteratorVariable,
RepeatIteratorVariable,
)
from .lazy import LazyVariableTracker
from .lists import (
BaseListVariable,
ListIteratorVariable,
ListVariable,
NamedTupleVariable,
RangeVariable,
RestrictedListSubclassVariable,
SliceVariable,
TupleVariable,
)
from .misc import (
AutogradFunctionContextVariable,
AutogradFunctionVariable,
ClosureVariable,
DeletedVariable,
GetAttrVariable,
InspectSignatureVariable,
LambdaVariable,
NewCellVariable,
NewGlobalVariable,
NumpyVariable,
PythonModuleVariable,
SkipFilesVariable,
SuperVariable,
UnknownVariable,
)
from .nn_module import NNModuleVariable, UnspecializedNNModuleVariable
from .tensor import (
FakeItemVariable,
NumpyNdarrayVariable,
SymNodeVariable,
TensorVariable,
UnspecializedPythonVariable,
)
from .torch import (
TorchCtxManagerClassVariable,
TorchInGraphFunctionVariable,
TorchVariable,
)
from .user_defined import UserDefinedClassVariable, UserDefinedObjectVariable
__all__ = [
"AutogradFunctionContextVariable",
"AutogradFunctionVariable",
"BaseListVariable",
"BuiltinVariable",
"ClosureVariable",
"ConstantVariable",
"ConstDictVariable",
"ContextWrappingVariable",
"CountIteratorVariable",
"CustomizedDictVariable",
"CycleIteratorVariable",
"DataClassVariable",
"DefaultDictVariable",
"DeletedVariable",
"DeterministicAlgorithmsVariable",
"EnumVariable",
"FakeItemVariable",
"GetAttrVariable",
"GradModeVariable",
"InspectSignatureVariable",
"IteratorVariable",
"LambdaVariable",
"LazyVariableTracker",
"ListIteratorVariable",
"ListVariable",
"NamedTupleVariable",
"NestedUserFunctionVariable",
"NewCellVariable",
"NewGlobalVariable",
"NNModuleVariable",
"NumpyNdarrayVariable",
"NumpyVariable",
"PythonModuleVariable",
"RangeVariable",
"RepeatIteratorVariable",
"RestrictedListSubclassVariable",
"SkipFilesVariable",
"SliceVariable",
"SuperVariable",
"TensorVariable",
"TorchCtxManagerClassVariable",
"TorchInGraphFunctionVariable",
"TorchVariable",
"TupleVariable",
"UnknownVariable",
"UnspecializedNNModuleVariable",
"UnspecializedPythonVariable",
"UserDefinedClassVariable",
"UserDefinedObjectVariable",
"UserFunctionVariable",
"UserMethodVariable",
"VariableTracker",
"WithExitFunctionVariable",
]
Back to Directory
File Manager