icefall/.mypy_cache/3.9/torch/distributed/rpc/options.data.json

{".class": "MypyFile", "_fullname": "torch.distributed.rpc.options", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "DeviceType": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.distributed.rpc.options.DeviceType", "line": 9, "no_args": false, "normalized": false, "target": {".class": "UnionType", "items": ["builtins.int", "builtins.str", "torch._C.device"]}}}, "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef"}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "TensorPipeRpcBackendOptions": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch._C._distributed_rpc._TensorPipeRpcBackendOptionsBase"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions", "name": "TensorPipeRpcBackendOptions", "type_vars": []}, "flags": [], "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions", "metaclass_type": null, "metadata": {}, "module_name": "torch.distributed.rpc.options", "mro": ["torch.distributed.rpc.options.TensorPipeRpcBackendOptions", "torch._C._distributed_rpc._TensorPipeRpcBackendOptionsBase", "torch._C._distributed_rpc.RpcBackendOptions", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 5, 5, 5, 5, 5, 5, 5], "arg_names": ["self", "num_worker_threads", "rpc_timeout", "init_method", "device_maps", "devices", "_transports", "_channels"], "flags": [], "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions.__init__", "name": "__init__", "type": {".class": "CallableType", "arg_kinds": [0, 5, 5, 5, 5, 5, 5, 5], "arg_names": ["self", "num_worker_threads", "rpc_timeout", "init_method", "device_maps", "devices", "_transports", "_channels"], "arg_types": ["torch.distributed.rpc.options.TensorPipeRpcBackendOptions", "builtins.int", "builtins.float", "builtins.str", {".class": "UnionType", "items": [{".class": "Instance", "args": ["builtins.str", {".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.dict"}], "type_ref": "builtins.dict"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.list"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "type_ref": "builtins.list"}, {".class": "NoneType"}]}, {".class": "UnionType", "items": [{".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 4}], "type_ref": "builtins.list"}, {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "__init__ of TensorPipeRpcBackendOptions", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "devices": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions.devices", "name": "devices", "type": {".class": "Instance", "args": ["torch._C.device"], "type_ref": "builtins.list"}}}, "set_device_map": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "to", "device_map"], "flags": [], "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions.set_device_map", "name": "set_device_map", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["self", "to", "device_map"], "arg_types": ["torch.distributed.rpc.options.TensorPipeRpcBackendOptions", "builtins.str", {".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.dict"}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "set_device_map of TensorPipeRpcBackendOptions", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "set_devices": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "devices"], "flags": [], "fullname": "torch.distributed.rpc.options.TensorPipeRpcBackendOptions.set_devices", "name": "set_devices", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "devices"], "arg_types": ["torch.distributed.rpc.options.TensorPipeRpcBackendOptions", {".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.list"}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "set_devices of TensorPipeRpcBackendOptions", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef"}, "_TensorPipeRpcBackendOptionsBase": {".class": "SymbolTableNode", "cross_ref": "torch._C._distributed_rpc._TensorPipeRpcBackendOptionsBase", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.distributed.rpc.options.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.distributed.rpc.options.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.distributed.rpc.options.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.distributed.rpc.options.__package__", "name": "__package__", "type": "builtins.str"}}, "_to_device": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["device"], "flags": [], "fullname": "torch.distributed.rpc.options._to_device", "name": "_to_device", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["device"], "arg_types": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_to_device", "ret_type": "torch._C.device", "type_guard": null, "variables": []}}}, "_to_device_list": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["devices"], "flags": [], "fullname": "torch.distributed.rpc.options._to_device_list", "name": "_to_device_list", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["devices"], "arg_types": [{".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.list"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_to_device_list", "ret_type": {".class": "Instance", "args": ["torch._C.device"], "type_ref": "builtins.list"}, "type_guard": null, "variables": []}}}, "_to_device_map": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["device_map"], "flags": [], "fullname": "torch.distributed.rpc.options._to_device_map", "name": "_to_device_map", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["device_map"], "arg_types": [{".class": "Instance", "args": [{".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}, {".class": "TypeAliasType", "args": [], "type_ref": "torch.distributed.rpc.options.DeviceType"}], "type_ref": "builtins.dict"}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_to_device_map", "ret_type": {".class": "Instance", "args": ["torch._C.device", "torch._C.device"], "type_ref": "builtins.dict"}, "type_guard": null, "variables": []}}}, "rpc_contants": {".class": "SymbolTableNode", "cross_ref": "torch.distributed.rpc.constants", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/distributed/rpc/options.py"}