icefall/.mypy_cache/3.9/torch/nn/utils/parametrize.data.json

{".class": "MypyFile", "_fullname": "torch.nn.utils.parametrize", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef"}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef"}, "ModuleDict": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.container.ModuleDict", "kind": "Gdef"}, "ModuleList": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.container.ModuleList", "kind": "Gdef"}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "Parameter": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parameter.Parameter", "kind": "Gdef"}, "ParametrizationList": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.container.ModuleList"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.utils.parametrize.ParametrizationList", "name": "ParametrizationList", "type_vars": []}, "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.utils.parametrize", "mro": ["torch.nn.utils.parametrize.ParametrizationList", "torch.nn.modules.container.ModuleList", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["self", "modules", "original", "unsafe"], "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["self", "modules", "original", "unsafe"], "arg_types": ["torch.nn.utils.parametrize.ParametrizationList", {".class": "Instance", "args": ["torch.nn.modules.module.Module"], "type_ref": "typing.Sequence"}, {".class": "UnionType", "items": ["torch._tensor.Tensor", "torch.nn.parameter.Parameter"]}, "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of ParametrizationList", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.utils.parametrize.ParametrizationList"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of ParametrizationList", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "is_tensor": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList.is_tensor", "name": "is_tensor", "type": "builtins.bool"}}, "ntensors": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList.ntensors", "name": "ntensors", "type": "builtins.int"}}, "original": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.utils.parametrize.ParametrizationList.original", "name": "original", "type": "torch._tensor.Tensor"}}, "right_inverse": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "flags": [], "fullname": "torch.nn.utils.parametrize.ParametrizationList.right_inverse", "name": "right_inverse", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "arg_types": ["torch.nn.utils.parametrize.ParametrizationList", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "right_inverse of ParametrizationList", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "unsafe": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.utils.parametrize.ParametrizationList.unsafe", "name": "unsafe", "type": "builtins.bool"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Sequence": {".class": "SymbolTableNode", "cross_ref": "typing.Sequence", "kind": "Gdef"}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef"}, "Tuple": {".class": "SymbolTableNode", "cross_ref": "typing.Tuple", "kind": "Gdef"}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize.__package__", "name": "__package__", "type": "builtins.str"}}, "_cache": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize._cache", "name": "_cache", "type": {".class": "Instance", "args": [{".class": "TupleType", "implicit": false, "items": ["builtins.int", "builtins.str"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "type_ref": "builtins.dict"}}}, "_cache_enabled": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.utils.parametrize._cache_enabled", "name": "_cache_enabled", "type": "builtins.int"}}, "_inject_new_class": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["module"], "flags": [], "fullname": "torch.nn.utils.parametrize._inject_new_class", "name": "_inject_new_class", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["module"], "arg_types": ["torch.nn.modules.module.Module"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_inject_new_class", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_inject_property": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["module", "tensor_name"], "flags": [], "fullname": "torch.nn.utils.parametrize._inject_property", "name": "_inject_property", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["module", "tensor_name"], "arg_types": ["torch.nn.modules.module.Module", "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_inject_property", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_register_parameter_or_buffer": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["module", "name", "X"], "flags": [], "fullname": "torch.nn.utils.parametrize._register_parameter_or_buffer", "name": "_register_parameter_or_buffer", "type": null}}, "cached": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": ["is_generator", "is_decorated"], "fullname": "torch.nn.utils.parametrize.cached", "name": "cached", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "cached", "type": {".class": "CallableType", "arg_kinds": [], "arg_names": [], "arg_types": [], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "cached", "ret_type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_of_any": 7}], "type_ref": "contextlib._GeneratorContextManager"}, "type_guard": null, "variables": []}}}}, "collections": {".class": "SymbolTableNode", "cross_ref": "collections", "kind": "Gdef"}, "contextmanager": {".class": "SymbolTableNode", "cross_ref": "contextlib.contextmanager", "kind": "Gdef"}, "is_parametrized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["module", "tensor_name"], "flags": [], "fullname": "torch.nn.utils.parametrize.is_parametrized", "name": "is_parametrized", "type": {".class": "CallableType", "arg_kinds": [0, 1], "arg_names": ["module", "tensor_name"], "arg_types": ["torch.nn.modules.module.Module", {".class": "UnionType", "items": ["builtins.str", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "is_parametrized", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "register_parametrization": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 5], "arg_names": ["module", "tensor_name", "parametrization", "unsafe"], "flags": [], "fullname": "torch.nn.utils.parametrize.register_parametrization", "name": "register_parametrization", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 5], "arg_names": ["module", "tensor_name", "parametrization", "unsafe"], "arg_types": ["torch.nn.modules.module.Module", "builtins.str", "torch.nn.modules.module.Module", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "register_parametrization", "ret_type": "torch.nn.modules.module.Module", "type_guard": null, "variables": []}}}, "remove_parametrizations": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["module", "tensor_name", "leave_parametrized"], "flags": [], "fullname": "torch.nn.utils.parametrize.remove_parametrizations", "name": "remove_parametrizations", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["module", "tensor_name", "leave_parametrized"], "arg_types": ["torch.nn.modules.module.Module", "builtins.str", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "remove_parametrizations", "ret_type": "torch.nn.modules.module.Module", "type_guard": null, "variables": []}}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/utils/parametrize.py"}