icefall/.mypy_cache/3.9/torch/nn/modules/rnn.data.json

{".class": "MypyFile", "_fullname": "torch.nn.modules.rnn", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "<subclass of \"float\" and \"Number\">": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.float", "numbers.Number"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">", "name": "<subclass of \"float\" and \"Number\">", "type_vars": []}, "flags": ["is_intersection"], "fullname": "torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">", "builtins.float", "numbers.Number", "builtins.object"], "names": {".class": "SymbolTable"}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "<subclass of \"float\" and \"Number\">1": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.float", "numbers.Number"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">1", "name": "<subclass of \"float\" and \"Number\">", "type_vars": []}, "flags": ["is_intersection"], "fullname": "torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">1", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.<subclass of \"float\" and \"Number\">1", "builtins.float", "numbers.Number", "builtins.object"], "names": {".class": "SymbolTable"}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "GRU": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.GRU", "name": "GRU", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.GRU", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.GRU", "torch.nn.modules.rnn.RNNBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.nn.modules.rnn.GRU.__init__", "name": "__init__", "type": null}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "OverloadedFuncDef", "flags": [], "fullname": "torch.nn.modules.rnn.GRU.forward", "impl": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload"], "fullname": "torch.nn.modules.rnn.GRU.forward", "name": "forward", "type": null}, "items": [{".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload", "is_decorated"], "fullname": "torch.nn.modules.rnn.GRU.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.GRU", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of GRU", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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": []}}, "is_overload": true, "var": {".class": "Var", "flags": [], "fullname": null, "name": "forward", "type": null}}, {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload", "is_decorated"], "fullname": "torch.nn.modules.rnn.GRU.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.GRU", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of GRU", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, "torch._tensor.Tensor"], "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": []}}, "is_overload": true, "var": {".class": "Var", "flags": [], "fullname": null, "name": "forward", "type": null}}], "type": {".class": "Overloaded", "items": [{".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.GRU", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of GRU", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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": []}, {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.GRU", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of GRU", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, "torch._tensor.Tensor"], "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_type": null, "type_vars": [], "typeddict_type": null}}, "GRUCell": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNCellBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.GRUCell", "name": "GRUCell", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.GRUCell", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.GRUCell", "torch.nn.modules.rnn.RNNCellBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.rnn.GRUCell.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.rnn.GRUCell", "builtins.int", "builtins.int", "builtins.bool", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of GRUCell", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": [], "fullname": "torch.nn.modules.rnn.GRUCell.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.GRUCell", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of GRUCell", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "LSTM": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.LSTM", "name": "LSTM", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.LSTM", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.LSTM", "torch.nn.modules.rnn.RNNBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTM.__init__", "name": "__init__", "type": null}}, "check_forward_args": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "input", "hidden", "batch_sizes"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTM.check_forward_args", "name": "check_forward_args", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "input", "hidden", "batch_sizes"], "arg_types": ["torch.nn.modules.rnn.LSTM", "torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "check_forward_args of LSTM", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "OverloadedFuncDef", "flags": [], "fullname": "torch.nn.modules.rnn.LSTM.forward", "impl": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload"], "fullname": "torch.nn.modules.rnn.LSTM.forward", "name": "forward", "type": null}, "items": [{".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload", "is_decorated"], "fullname": "torch.nn.modules.rnn.LSTM.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.LSTM", "torch._tensor.Tensor", {".class": "UnionType", "items": [{".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}], "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": []}}, "is_overload": true, "var": {".class": "Var", "flags": [], "fullname": null, "name": "forward", "type": null}}, {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": ["is_overload", "is_decorated"], "fullname": "torch.nn.modules.rnn.LSTM.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.LSTM", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "UnionType", "items": [{".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}], "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": []}}, "is_overload": true, "var": {".class": "Var", "flags": [], "fullname": null, "name": "forward", "type": null}}], "type": {".class": "Overloaded", "items": [{".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.LSTM", "torch._tensor.Tensor", {".class": "UnionType", "items": [{".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}], "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": []}, {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.LSTM", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "UnionType", "items": [{".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}, {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}], "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": []}]}}}, "get_expected_cell_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTM.get_expected_cell_size", "name": "get_expected_cell_size", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "arg_types": ["torch.nn.modules.rnn.LSTM", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "get_expected_cell_size of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": ["builtins.int", "builtins.int", "builtins.int"], "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": []}}}, "permute_hidden": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "hx", "permutation"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTM.permute_hidden", "name": "permute_hidden", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "hx", "permutation"], "arg_types": ["torch.nn.modules.rnn.LSTM", {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "permute_hidden of LSTM", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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_type": null, "type_vars": [], "typeddict_type": null}}, "LSTMCell": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNCellBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.LSTMCell", "name": "LSTMCell", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.LSTMCell", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.LSTMCell", "torch.nn.modules.rnn.RNNCellBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTMCell.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.rnn.LSTMCell", "builtins.int", "builtins.int", "builtins.bool", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of LSTMCell", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": [], "fullname": "torch.nn.modules.rnn.LSTMCell.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.LSTMCell", "torch._tensor.Tensor", {".class": "UnionType", "items": [{".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of LSTMCell", "ret_type": {".class": "TupleType", "implicit": false, "items": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "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_type": null, "type_vars": [], "typeddict_type": null}}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef"}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "PackedSequence": {".class": "SymbolTableNode", "cross_ref": "torch.nn.utils.rnn.PackedSequence", "kind": "Gdef"}, "Parameter": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parameter.Parameter", "kind": "Gdef"}, "RNN": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.RNN", "name": "RNN", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.RNN", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.RNN", "torch.nn.modules.rnn.RNNBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.nn.modules.rnn.RNN.__init__", "name": "__init__", "type": null}}, "nonlinearity": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNN.nonlinearity", "name": "nonlinearity", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "RNNBase": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.module.Module"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.RNNBase", "name": "RNNBase", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.RNNBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__constants__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.rnn.RNNBase.__constants__", "name": "__constants__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "mode", "input_size", "hidden_size", "num_layers", "bias", "batch_first", "dropout", "bidirectional", "proj_size", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "mode", "input_size", "hidden_size", "num_layers", "bias", "batch_first", "dropout", "bidirectional", "proj_size", "device", "dtype"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "builtins.str", "builtins.int", "builtins.int", "builtins.int", "builtins.bool", "builtins.bool", "builtins.float", "builtins.bool", "builtins.int", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of RNNBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "__jit_unused_properties__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.rnn.RNNBase.__jit_unused_properties__", "name": "__jit_unused_properties__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__setattr__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "attr", "value"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.__setattr__", "name": "__setattr__", "type": null}}, "__setstate__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "d"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.__setstate__", "name": "__setstate__", "type": null}}, "_all_weights": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase._all_weights", "name": "_all_weights", "type": {".class": "Instance", "args": [{".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}], "type_ref": "builtins.list"}}}, "_apply": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "fn"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase._apply", "name": "_apply", "type": null}}, "_flat_weights": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase._flat_weights", "name": "_flat_weights", "type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_of_any": 7}], "type_ref": "builtins.list"}}}, "_flat_weights_names": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase._flat_weights_names", "name": "_flat_weights_names", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "_replicate_for_data_parallel": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase._replicate_for_data_parallel", "name": "_replicate_for_data_parallel", "type": null}}, "all_weights": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": ["is_property", "is_decorated"], "fullname": "torch.nn.modules.rnn.RNNBase.all_weights", "name": "all_weights", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "all_weights of RNNBase", "ret_type": {".class": "Instance", "args": [{".class": "Instance", "args": ["torch.nn.parameter.Parameter"], "type_ref": "builtins.list"}], "type_ref": "builtins.list"}, "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_property", "is_ready"], "fullname": null, "name": "all_weights", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "all_weights of RNNBase", "ret_type": {".class": "Instance", "args": [{".class": "Instance", "args": ["torch.nn.parameter.Parameter"], "type_ref": "builtins.list"}], "type_ref": "builtins.list"}, "type_guard": null, "variables": []}}}}, "batch_first": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.batch_first", "name": "batch_first", "type": "builtins.bool"}}, "bias": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.bias", "name": "bias", "type": "builtins.bool"}}, "bidirectional": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.bidirectional", "name": "bidirectional", "type": "builtins.bool"}}, "check_forward_args": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "input", "hidden", "batch_sizes"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.check_forward_args", "name": "check_forward_args", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0], "arg_names": ["self", "input", "hidden", "batch_sizes"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "check_forward_args of RNNBase", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "check_hidden_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["self", "hx", "expected_hidden_size", "msg"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.check_hidden_size", "name": "check_hidden_size", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["self", "hx", "expected_hidden_size", "msg"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": ["builtins.int", "builtins.int", "builtins.int"], "partial_fallback": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 6}], "type_ref": "builtins.tuple"}}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "check_hidden_size of RNNBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "check_input": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.check_input", "name": "check_input", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "check_input of RNNBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "dropout": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.dropout", "name": "dropout", "type": "builtins.float"}}, "extra_repr": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of RNNBase", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "flatten_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.flatten_parameters", "name": "flatten_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "flatten_parameters of RNNBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.RNNBase", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}]}, {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of RNNBase", "ret_type": {".class": "TupleType", "implicit": false, "items": [{".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "TupleType", "implicit": false, "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "partial_fallback": "torch.nn.utils.rnn.PackedSequence"}]}, "torch._tensor.Tensor"], "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": []}}}, "get_expected_hidden_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.get_expected_hidden_size", "name": "get_expected_hidden_size", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input", "batch_sizes"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "get_expected_hidden_size of RNNBase", "ret_type": {".class": "TupleType", "implicit": false, "items": ["builtins.int", "builtins.int", "builtins.int"], "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": []}}}, "hidden_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.hidden_size", "name": "hidden_size", "type": "builtins.int"}}, "input_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.input_size", "name": "input_size", "type": "builtins.int"}}, "mode": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.mode", "name": "mode", "type": "builtins.str"}}, "num_layers": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.num_layers", "name": "num_layers", "type": "builtins.int"}}, "permute_hidden": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "hx", "permutation"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.permute_hidden", "name": "permute_hidden", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "hx", "permutation"], "arg_types": ["torch.nn.modules.rnn.RNNBase", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "permute_hidden of RNNBase", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "proj_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNBase.proj_size", "name": "proj_size", "type": "builtins.int"}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNBase.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of RNNBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "RNNCell": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.rnn.RNNCellBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.RNNCell", "name": "RNNCell", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.RNNCell", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.RNNCell", "torch.nn.modules.rnn.RNNCellBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__constants__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.rnn.RNNCell.__constants__", "name": "__constants__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "nonlinearity", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNCell.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "nonlinearity", "device", "dtype"], "arg_types": ["torch.nn.modules.rnn.RNNCell", "builtins.int", "builtins.int", "builtins.bool", "builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of RNNCell", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNCell.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["self", "input", "hx"], "arg_types": ["torch.nn.modules.rnn.RNNCell", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of RNNCell", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "nonlinearity": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCell.nonlinearity", "name": "nonlinearity", "type": "builtins.str"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "RNNCellBase": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.module.Module"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.rnn.RNNCellBase", "name": "RNNCellBase", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.rnn", "mro": ["torch.nn.modules.rnn.RNNCellBase", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__constants__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.rnn.RNNCellBase.__constants__", "name": "__constants__", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "builtins.list"}}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "num_chunks", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 0, 1, 1], "arg_names": ["self", "input_size", "hidden_size", "bias", "num_chunks", "device", "dtype"], "arg_types": ["torch.nn.modules.rnn.RNNCellBase", "builtins.int", "builtins.int", "builtins.bool", "builtins.int", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of RNNCellBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "bias": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCellBase.bias", "name": "bias", "type": "builtins.bool"}}, "bias_hh": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase.bias_hh", "name": "bias_hh", "type": "torch.nn.parameter.Parameter"}}, "bias_ih": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase.bias_ih", "name": "bias_ih", "type": "torch.nn.parameter.Parameter"}}, "extra_repr": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNCellBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of RNNCellBase", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "hidden_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCellBase.hidden_size", "name": "hidden_size", "type": "builtins.int"}}, "input_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCellBase.input_size", "name": "input_size", "type": "builtins.int"}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.rnn.RNNCellBase.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.rnn.RNNCellBase"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of RNNCellBase", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "weight_hh": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCellBase.weight_hh", "name": "weight_hh", "type": "torch._tensor.Tensor"}}, "weight_ih": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.rnn.RNNCellBase.weight_ih", "name": "weight_ih", "type": "torch._tensor.Tensor"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "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"}, "_VF": {".class": "SymbolTableNode", "cross_ref": "torch._VF", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.rnn.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.rnn.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.rnn.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.rnn.__package__", "name": "__package__", "type": "builtins.str"}}, "_rnn_impls": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.rnn._rnn_impls", "name": "_rnn_impls", "type": {".class": "Instance", "args": ["builtins.str", {".class": "AnyType", "missing_import_name": null, "source_any": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 5}, "type_of_any": 7}], "type_ref": "builtins.dict"}}}, "apply_permutation": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["tensor", "permutation", "dim"], "flags": [], "fullname": "torch.nn.modules.rnn.apply_permutation", "name": "apply_permutation", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["tensor", "permutation", "dim"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "apply_permutation", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "cast": {".class": "SymbolTableNode", "cross_ref": "typing.cast", "kind": "Gdef"}, "init": {".class": "SymbolTableNode", "cross_ref": "torch.nn.init", "kind": "Gdef"}, "math": {".class": "SymbolTableNode", "cross_ref": "math", "kind": "Gdef"}, "numbers": {".class": "SymbolTableNode", "cross_ref": "numbers", "kind": "Gdef"}, "overload": {".class": "SymbolTableNode", "cross_ref": "typing.overload", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/modules/rnn.py"}