icefall/.mypy_cache/3.9/torch/nn/parallel/distributed.data.json

{".class": "MypyFile", "_fullname": "torch.nn.parallel.distributed", "is_partial_stub_package": false, "is_stub": true, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef", "module_hidden": true, "module_public": false}, "DistributedDataParallel": {".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.parallel.distributed.DistributedDataParallel", "name": "DistributedDataParallel", "type_vars": []}, "flags": [], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel", "metaclass_type": null, "metadata": {}, "module_name": "torch.nn.parallel.distributed", "mro": ["torch.nn.parallel.distributed.DistributedDataParallel", "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, 1, 1, 1, 1, 1], "arg_names": ["self", "module", "device_ids", "output_device", "dim", "broadcast_buffers", "process_group", "bucket_cap_mb", "find_unused_parameters", "check_reduction"], "flags": [], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["self", "module", "device_ids", "output_device", "dim", "broadcast_buffers", "process_group", "bucket_cap_mb", "find_unused_parameters", "check_reduction"], "arg_types": ["torch.nn.parallel.distributed.DistributedDataParallel", "torch.nn.modules.module.Module", {".class": "UnionType", "items": [{".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.parallel.common_types._device_t"}], "type_ref": "typing.Sequence"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.int", "torch._C.device", {".class": "NoneType"}]}, "builtins.int", "builtins.bool", {".class": "UnionType", "items": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "NoneType"}]}, "builtins.float", "builtins.bool", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of DistributedDataParallel", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "broadcast_bucket_size": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.broadcast_bucket_size", "name": "broadcast_bucket_size", "type": "builtins.float"}}, "broadcast_buffers": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.broadcast_buffers", "name": "broadcast_buffers", "type": "builtins.bool"}}, "bucket_bytes_cap": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.bucket_bytes_cap", "name": "bucket_bytes_cap", "type": "builtins.float"}}, "check_reduction": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.check_reduction", "name": "check_reduction", "type": "builtins.bool"}}, "device_ids": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.device_ids", "name": "device_ids", "type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.parallel.common_types._devices_t"}}}, "dim": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.dim", "name": "dim", "type": "builtins.int"}}, "module": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.module", "name": "module", "type": "torch.nn.modules.module.Module"}}, "output_device": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.output_device", "name": "output_device", "type": {".class": "TypeAliasType", "args": [], "type_ref": "torch.nn.parallel.common_types._device_t"}}}, "process_group": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class", "is_ready"], "fullname": "torch.nn.parallel.distributed.DistributedDataParallel.process_group", "name": "process_group", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef", "module_hidden": true, "module_public": false}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef", "module_hidden": true, "module_public": false}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.parallel.distributed.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.parallel.distributed.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.parallel.distributed.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.parallel.distributed.__package__", "name": "__package__", "type": "builtins.str"}}, "_device_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parallel.common_types._device_t", "kind": "Gdef", "module_hidden": true, "module_public": false}, "_devices_t": {".class": "SymbolTableNode", "cross_ref": "torch.nn.parallel.common_types._devices_t", "kind": "Gdef", "module_hidden": true, "module_public": false}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/parallel/distributed.pyi"}