icefall/.mypy_cache/3.9/torch/utils/_pytree.data.json

{".class": "MypyFile", "_fullname": "torch.utils._pytree", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef"}, "Callable": {".class": "SymbolTableNode", "cross_ref": "typing.Callable", "kind": "Gdef"}, "Context": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.utils._pytree.Context", "line": 28, "no_args": false, "normalized": false, "target": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}}}, "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef"}, "FlattenFunc": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.utils._pytree.FlattenFunc", "line": 30, "no_args": false, "normalized": false, "target": {".class": "CallableType", "arg_kinds": [0], "arg_names": [null], "arg_types": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": null, "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "LeafSpec": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.utils._pytree.TreeSpec"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.utils._pytree.LeafSpec", "name": "LeafSpec", "type_vars": []}, "flags": [], "fullname": "torch.utils._pytree.LeafSpec", "metaclass_type": null, "metadata": {}, "module_name": "torch.utils._pytree", "mro": ["torch.utils._pytree.LeafSpec", "torch.utils._pytree.TreeSpec", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.utils._pytree.LeafSpec.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.utils._pytree.LeafSpec"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of LeafSpec", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "__repr__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.utils._pytree.LeafSpec.__repr__", "name": "__repr__", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": [null], "arg_types": ["torch.utils._pytree.LeafSpec"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__repr__ of LeafSpec", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "NamedTuple": {".class": "SymbolTableNode", "cross_ref": "typing.NamedTuple", "kind": "Gdef"}, "NodeDef": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": [{".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.utils._pytree.NodeDef", "name": "NodeDef", "type_vars": []}, "flags": ["is_named_tuple"], "fullname": "torch.utils._pytree.NodeDef", "metaclass_type": null, "metadata": {"namedtuple": {"fields": ["flatten_fn", "unflatten_fn"]}}, "module_name": "torch.utils._pytree", "mro": ["torch.utils._pytree.NodeDef", "builtins.tuple", "typing.Sequence", "typing.Collection", "typing.Reversible", "typing.Iterable", "typing.Container", "builtins.object"], "names": {".class": "SymbolTable", "_NT": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "TypeVarExpr", "fullname": "torch.utils._pytree.NodeDef._NT", "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}}, "__annotations__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef.__annotations__", "name": "__annotations__", "type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.dict"}}}, "__doc__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__new__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["_cls", "flatten_fn", "unflatten_fn"], "flags": [], "fullname": "torch.utils._pytree.NodeDef.__new__", "name": "__new__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["_cls", "flatten_fn", "unflatten_fn"], "arg_types": [{".class": "TypeType", "item": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__new__ of NodeDef", "ret_type": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}, "type_guard": null, "variables": [{".class": "TypeVarDef", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}]}}, "plugin_generated": true}, "_asdict": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["_self"], "flags": [], "fullname": "torch.utils._pytree.NodeDef._asdict", "name": "_asdict", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["_self"], "arg_types": [{".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_asdict of NodeDef", "ret_type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.dict"}, "type_guard": null, "variables": [{".class": "TypeVarDef", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}]}}, "plugin_generated": true}, "_field_defaults": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef._field_defaults", "name": "_field_defaults", "type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.dict"}}}, "_field_types": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef._field_types", "name": "_field_types", "type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.dict"}}}, "_fields": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef._fields", "name": "_fields", "type": {".class": "TupleType", "implicit": false, "items": ["builtins.str", "builtins.str"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}}}, "_make": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 5, 5], "arg_names": ["_cls", "iterable", "new", "len"], "flags": ["is_class", "is_decorated"], "fullname": "torch.utils._pytree.NodeDef._make", "name": "_make", "type": {".class": "CallableType", "arg_kinds": [0, 0, 5, 5], "arg_names": ["_cls", "iterable", "new", "len"], "arg_types": [{".class": "TypeType", "item": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}}, {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "typing.Iterable"}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_make of NodeDef", "ret_type": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}, "type_guard": null, "variables": [{".class": "TypeVarDef", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}]}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_classmethod", "is_ready"], "fullname": "torch.utils._pytree.NodeDef._make", "name": "_make", "type": {".class": "CallableType", "arg_kinds": [0, 0, 5, 5], "arg_names": ["_cls", "iterable", "new", "len"], "arg_types": [{".class": "TypeType", "item": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}}, {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "typing.Iterable"}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_make of NodeDef", "ret_type": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}, "type_guard": null, "variables": [{".class": "TypeVarDef", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}]}}}, "plugin_generated": true}, "_replace": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 5, 5], "arg_names": ["_self", "flatten_fn", "unflatten_fn"], "flags": [], "fullname": "torch.utils._pytree.NodeDef._replace", "name": "_replace", "type": {".class": "CallableType", "arg_kinds": [0, 5, 5], "arg_names": ["_self", "flatten_fn", "unflatten_fn"], "arg_types": [{".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_replace of NodeDef", "ret_type": {".class": "TypeVarType", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}, "type_guard": null, "variables": [{".class": "TypeVarDef", "fullname": "torch.utils._pytree.NodeDef._NT", "id": -1, "name": "_NT", "upper_bound": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "values": [], "variance": 0}]}}, "plugin_generated": true}, "_source": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.utils._pytree.NodeDef._source", "name": "_source", "type": "builtins.str"}}, "flatten_fn": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_property", "is_ready"], "fullname": "torch.utils._pytree.NodeDef.flatten_fn", "name": "flatten_fn", "type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}}}, "flatten_fn-redefinition": {".class": "SymbolTableNode", "cross_ref": "torch.utils._pytree.NodeDef.flatten_fn", "kind": "Mdef"}, "unflatten_fn": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_property", "is_ready"], "fullname": "torch.utils._pytree.NodeDef.unflatten_fn", "name": "unflatten_fn", "type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}}}, "unflatten_fn-redefinition": {".class": "SymbolTableNode", "cross_ref": "torch.utils._pytree.NodeDef.unflatten_fn", "kind": "Mdef"}}, "tuple_type": {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": {".class": "Instance", "args": ["builtins.function"], "type_ref": "builtins.tuple"}}, "type_vars": [], "typeddict_type": null}}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "PyTree": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.utils._pytree.PyTree", "line": 29, "no_args": false, "normalized": false, "target": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}}}, "SUPPORTED_NODES": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.utils._pytree.SUPPORTED_NODES", "name": "SUPPORTED_NODES", "type": {".class": "Instance", "args": [{".class": "TypeType", "item": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}}, {".class": "TupleType", "implicit": false, "items": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "partial_fallback": "torch.utils._pytree.NodeDef"}], "type_ref": "builtins.dict"}}}, "TreeSpec": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.object"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.utils._pytree.TreeSpec", "name": "TreeSpec", "type_vars": []}, "flags": [], "fullname": "torch.utils._pytree.TreeSpec", "metaclass_type": null, "metadata": {}, "module_name": "torch.utils._pytree", "mro": ["torch.utils._pytree.TreeSpec", "builtins.object"], "names": {".class": "SymbolTable", "__eq__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "other"], "flags": [], "fullname": "torch.utils._pytree.TreeSpec.__eq__", "name": "__eq__", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": [null, null], "arg_types": ["torch.utils._pytree.TreeSpec", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__eq__ of TreeSpec", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "typ", "context", "children_specs"], "flags": [], "fullname": "torch.utils._pytree.TreeSpec.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "typ", "context", "children_specs"], "arg_types": ["torch.utils._pytree.TreeSpec", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}, {".class": "Instance", "args": ["torch.utils._pytree.TreeSpec"], "type_ref": "builtins.list"}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of TreeSpec", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "__ne__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "other"], "flags": [], "fullname": "torch.utils._pytree.TreeSpec.__ne__", "name": "__ne__", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": [null, null], "arg_types": ["torch.utils._pytree.TreeSpec", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__ne__ of TreeSpec", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "__repr__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.utils._pytree.TreeSpec.__repr__", "name": "__repr__", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": [null], "arg_types": ["torch.utils._pytree.TreeSpec"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__repr__ of TreeSpec", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "children_specs": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.utils._pytree.TreeSpec.children_specs", "name": "children_specs", "type": {".class": "Instance", "args": ["torch.utils._pytree.TreeSpec"], "type_ref": "builtins.list"}}}, "context": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.utils._pytree.TreeSpec.context", "name": "context", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}}}, "num_leaves": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": ["explicit_self_type", "is_ready"], "fullname": "torch.utils._pytree.TreeSpec.num_leaves", "name": "num_leaves", "type": "builtins.int"}}, "type": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.utils._pytree.TreeSpec.type", "name": "type", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Tuple": {".class": "SymbolTableNode", "cross_ref": "typing.Tuple", "kind": "Gdef"}, "Type": {".class": "SymbolTableNode", "cross_ref": "typing.Type", "kind": "Gdef"}, "UnflattenFunc": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.utils._pytree.UnflattenFunc", "line": 31, "no_args": false, "normalized": false, "target": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": [null, null], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": null, "ret_type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}, "type_guard": null, "variables": []}}}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.utils._pytree.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.utils._pytree.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.utils._pytree.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.utils._pytree.__package__", "name": "__package__", "type": "builtins.str"}}, "_broadcast_to_and_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["pytree", "spec"], "flags": [], "fullname": "torch.utils._pytree._broadcast_to_and_flatten", "name": "_broadcast_to_and_flatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["pytree", "spec"], "arg_types": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}, "torch.utils._pytree.TreeSpec"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_broadcast_to_and_flatten", "ret_type": {".class": "UnionType", "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "NoneType"}]}, "type_guard": null, "variables": []}}}, "_dict_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["d"], "flags": [], "fullname": "torch.utils._pytree._dict_flatten", "name": "_dict_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["d"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.dict"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_dict_flatten", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "_dict_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "flags": [], "fullname": "torch.utils._pytree._dict_unflatten", "name": "_dict_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_dict_unflatten", "ret_type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.dict"}, "type_guard": null, "variables": []}}}, "_get_node_type": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["pytree"], "flags": [], "fullname": "torch.utils._pytree._get_node_type", "name": "_get_node_type", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["pytree"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_get_node_type", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}, "_is_leaf": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["pytree"], "flags": [], "fullname": "torch.utils._pytree._is_leaf", "name": "_is_leaf", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["pytree"], "arg_types": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_is_leaf", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "_is_namedtuple_instance": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["pytree"], "flags": [], "fullname": "torch.utils._pytree._is_namedtuple_instance", "name": "_is_namedtuple_instance", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["pytree"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_is_namedtuple_instance", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "_list_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["d"], "flags": [], "fullname": "torch.utils._pytree._list_flatten", "name": "_list_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["d"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_list_flatten", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "_list_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "flags": [], "fullname": "torch.utils._pytree._list_unflatten", "name": "_list_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_list_unflatten", "ret_type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, "type_guard": null, "variables": []}}}, "_namedtuple_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["d"], "flags": [], "fullname": "torch.utils._pytree._namedtuple_flatten", "name": "_namedtuple_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["d"], "arg_types": ["typing.NamedTuple"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_namedtuple_flatten", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "_namedtuple_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "flags": [], "fullname": "torch.utils._pytree._namedtuple_unflatten", "name": "_namedtuple_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_namedtuple_unflatten", "ret_type": "typing.NamedTuple", "type_guard": null, "variables": []}}}, "_register_pytree_node": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["typ", "flatten_fn", "unflatten_fn"], "flags": [], "fullname": "torch.utils._pytree._register_pytree_node", "name": "_register_pytree_node", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["typ", "flatten_fn", "unflatten_fn"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.FlattenFunc"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.UnflattenFunc"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_register_pytree_node", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "_tuple_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["d"], "flags": [], "fullname": "torch.utils._pytree._tuple_flatten", "name": "_tuple_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["d"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.tuple"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_tuple_flatten", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "_tuple_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "flags": [], "fullname": "torch.utils._pytree._tuple_unflatten", "name": "_tuple_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["values", "context"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.Context"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_tuple_unflatten", "ret_type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.tuple"}, "type_guard": null, "variables": []}}}, "cast": {".class": "SymbolTableNode", "cross_ref": "typing.cast", "kind": "Gdef"}, "namedtuple": {".class": "SymbolTableNode", "cross_ref": "collections.namedtuple", "kind": "Gdef"}, "tree_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["pytree"], "flags": [], "fullname": "torch.utils._pytree.tree_flatten", "name": "tree_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["pytree"], "arg_types": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "tree_flatten", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, "torch.utils._pytree.TreeSpec"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "type_guard": null, "variables": []}}}, "tree_map": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["fn", "pytree"], "flags": [], "fullname": "torch.utils._pytree.tree_map", "name": "tree_map", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["fn", "pytree"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "tree_map", "ret_type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}, "type_guard": null, "variables": []}}}, "tree_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["values", "spec"], "flags": [], "fullname": "torch.utils._pytree.tree_unflatten", "name": "tree_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["values", "spec"], "arg_types": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.list"}, "torch.utils._pytree.TreeSpec"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "tree_unflatten", "ret_type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.utils._pytree.PyTree"}, "type_guard": null, "variables": []}}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/utils/_pytree.py"}