__init__.py 2.2 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253545556575859606162636465666768697071727374757677787980818283848586878889
  1. from .base import VariableTracker
  2. from .builtin import BuiltinVariable
  3. from .constant import ConstantVariable, EnumVariable
  4. from .dicts import ConstDictVariable, DataClassVariable, DefaultDictVariable
  5. from .functions import (
  6. NestedUserFunctionVariable,
  7. UserFunctionVariable,
  8. UserMethodVariable,
  9. )
  10. from .lists import (
  11. BaseListVariable,
  12. ListIteratorVariable,
  13. ListVariable,
  14. NamedTupleVariable,
  15. RangeVariable,
  16. SliceVariable,
  17. TupleVariable,
  18. )
  19. from .misc import (
  20. AutogradFunctionVariable,
  21. BlackHoleVariable,
  22. ClosureVariable,
  23. ContextWrappingVariable,
  24. CUDAStreamContextVariable,
  25. CUDAStreamVariable,
  26. GetAttrVariable,
  27. GradModeVariable,
  28. InspectSignatureVariable,
  29. LambdaVariable,
  30. NewCellVariable,
  31. NewGlobalVariable,
  32. NumpyVariable,
  33. PythonModuleVariable,
  34. SuperVariable,
  35. UnknownVariable,
  36. WithExitFunctionVariable,
  37. )
  38. from .nn_module import NNModuleVariable, UnspecializedNNModuleVariable
  39. from .tensor import (
  40. FakeItemVariable,
  41. SymNodeVariable,
  42. TensorVariable,
  43. UnspecializedPythonVariable,
  44. )
  45. from .torch import TorchVariable
  46. from .user_defined import UserDefinedClassVariable, UserDefinedObjectVariable
  47. __all__ = [
  48. "AutogradFunctionVariable",
  49. "BaseListVariable",
  50. "BlackHoleVariable",
  51. "BuiltinVariable",
  52. "ClosureVariable",
  53. "ConstantVariable",
  54. "ConstDictVariable",
  55. "ContextWrappingVariable",
  56. "DataClassVariable",
  57. "DefaultDictVariable",
  58. "EnumVariable",
  59. "FakeItemVariable",
  60. "GetAttrVariable",
  61. "GradModeVariable",
  62. "InspectSignatureVariable",
  63. "LambdaVariable",
  64. "ListIteratorVariable",
  65. "ListVariable",
  66. "NamedTupleVariable",
  67. "NestedUserFunctionVariable",
  68. "NewCellVariable",
  69. "NewGlobalVariable",
  70. "NNModuleVariable",
  71. "NumpyVariable",
  72. "PythonModuleVariable",
  73. "RangeVariable",
  74. "SliceVariable",
  75. "SuperVariable",
  76. "TensorVariable",
  77. "TorchVariable",
  78. "TupleVariable",
  79. "UnknownVariable",
  80. "UnspecializedNNModuleVariable",
  81. "UnspecializedPythonVariable",
  82. "UserDefinedClassVariable",
  83. "UserDefinedObjectVariable",
  84. "UserFunctionVariable",
  85. "UserMethodVariable",
  86. "VariableTracker",
  87. "WithExitFunctionVariable",
  88. ]