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

{".class": "MypyFile", "_fullname": "torch.nn.modules.linear", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Bilinear": {".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.linear.Bilinear", "name": "Bilinear", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.linear.Bilinear", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.linear", "mro": ["torch.nn.modules.linear.Bilinear", "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.linear.Bilinear.__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], "arg_names": ["self", "in1_features", "in2_features", "out_features", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.linear.Bilinear.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 1, 1, 1], "arg_names": ["self", "in1_features", "in2_features", "out_features", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.linear.Bilinear", "builtins.int", "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 Bilinear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "bias": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.linear.Bilinear.bias", "name": "bias", "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.linear.Bilinear.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.linear.Bilinear"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of Bilinear", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input1", "input2"], "flags": [], "fullname": "torch.nn.modules.linear.Bilinear.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "input1", "input2"], "arg_types": ["torch.nn.modules.linear.Bilinear", "torch._tensor.Tensor", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Bilinear", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "in1_features": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Bilinear.in1_features", "name": "in1_features", "type": "builtins.int"}}, "in2_features": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Bilinear.in2_features", "name": "in2_features", "type": "builtins.int"}}, "out_features": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Bilinear.out_features", "name": "out_features", "type": "builtins.int"}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.linear.Bilinear.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.linear.Bilinear"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of Bilinear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "weight": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Bilinear.weight", "name": "weight", "type": "torch._tensor.Tensor"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "F": {".class": "SymbolTableNode", "cross_ref": "torch.nn.functional", "kind": "Gdef"}, "Identity": {".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.linear.Identity", "name": "Identity", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.linear.Identity", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.linear", "mro": ["torch.nn.modules.linear.Identity", "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.linear.Identity.__init__", "name": "__init__", "type": null}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.nn.modules.linear.Identity.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "arg_types": ["torch.nn.modules.linear.Identity", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Identity", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "LazyLinear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.lazy.LazyModuleMixin", "torch.nn.modules.linear.Linear"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.linear.LazyLinear", "name": "LazyLinear", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.linear.LazyLinear", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.linear", "mro": ["torch.nn.modules.linear.LazyLinear", "torch.nn.modules.lazy.LazyModuleMixin", "torch.nn.modules.linear.Linear", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["self", "out_features", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.linear.LazyLinear.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["self", "out_features", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.linear.LazyLinear", "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 LazyLinear", "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.linear.LazyLinear.bias", "name": "bias", "type": "torch.nn.parameter.UninitializedParameter"}}, "cls_to_become": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.nn.modules.linear.LazyLinear.cls_to_become", "name": "cls_to_become", "type": null}}, "initialize_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.nn.modules.linear.LazyLinear.initialize_parameters", "name": "initialize_parameters", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "arg_types": ["torch.nn.modules.linear.LazyLinear", {".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": "initialize_parameters of LazyLinear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.linear.LazyLinear.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.linear.LazyLinear"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of LazyLinear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "weight": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.LazyLinear.weight", "name": "weight", "type": "torch.nn.parameter.UninitializedParameter"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "LazyModuleMixin": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.lazy.LazyModuleMixin", "kind": "Gdef"}, "Linear": {".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.linear.Linear", "name": "Linear", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.linear.Linear", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.linear", "mro": ["torch.nn.modules.linear.Linear", "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.linear.Linear.__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], "arg_names": ["self", "in_features", "out_features", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.linear.Linear.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "in_features", "out_features", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.linear.Linear", "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 Linear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "bias": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.nn.modules.linear.Linear.bias", "name": "bias", "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.linear.Linear.extra_repr", "name": "extra_repr", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.linear.Linear"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "extra_repr of Linear", "ret_type": "builtins.str", "type_guard": null, "variables": []}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.nn.modules.linear.Linear.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "input"], "arg_types": ["torch.nn.modules.linear.Linear", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Linear", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "in_features": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Linear.in_features", "name": "in_features", "type": "builtins.int"}}, "out_features": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Linear.out_features", "name": "out_features", "type": "builtins.int"}}, "reset_parameters": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.nn.modules.linear.Linear.reset_parameters", "name": "reset_parameters", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.nn.modules.linear.Linear"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "reset_parameters of Linear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "weight": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.modules.linear.Linear.weight", "name": "weight", "type": "torch._tensor.Tensor"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef"}, "NonDynamicallyQuantizableLinear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.nn.modules.linear.Linear"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.nn.modules.linear.NonDynamicallyQuantizableLinear", "name": "NonDynamicallyQuantizableLinear", "type_vars": []}, "flags": [], "fullname": "torch.nn.modules.linear.NonDynamicallyQuantizableLinear", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.modules.linear", "mro": ["torch.nn.modules.linear.NonDynamicallyQuantizableLinear", "torch.nn.modules.linear.Linear", "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", "in_features", "out_features", "bias", "device", "dtype"], "flags": [], "fullname": "torch.nn.modules.linear.NonDynamicallyQuantizableLinear.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["self", "in_features", "out_features", "bias", "device", "dtype"], "arg_types": ["torch.nn.modules.linear.NonDynamicallyQuantizableLinear", "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 NonDynamicallyQuantizableLinear", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Parameter": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parameter.Parameter", "kind": "Gdef"}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef"}, "UninitializedParameter": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parameter.UninitializedParameter", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.linear.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.linear.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.linear.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.modules.linear.__package__", "name": "__package__", "type": "builtins.str"}}, "init": {".class": "SymbolTableNode", "cross_ref": "torch.nn.init", "kind": "Gdef"}, "math": {".class": "SymbolTableNode", "cross_ref": "math", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/modules/linear.py"}