{".class":"MypyFile","_fullname":"torch.distributed.tensor.parallel._data_parallel_utils","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","DTensorSpec":{".class":"SymbolTableNode","cross_ref":"torch.distributed._tensor.placement_types.DTensorSpec","kind":"Gdef"},"DistributedTensor":{".class":"SymbolTableNode","cross_ref":"torch.distributed._tensor.api.DTensor","kind":"Gdef"},"Optional":{".class":"SymbolTableNode","cross_ref":"typing.Optional","kind":"Gdef"},"Tuple":{".class":"SymbolTableNode","cross_ref":"typing.Tuple","kind":"Gdef"},"__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils.__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":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils.__doc__","name":"__doc__","type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils.__file__","name":"__file__","type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils.__name__","name":"__name__","type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils.__package__","name":"__package__","type":"builtins.str"}},"_flatten_tensor":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["tensor"],"dataclass_transform_spec":null,"flags":[],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils._flatten_tensor","name":"_flatten_tensor","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["tensor"],"arg_types":["torch._tensor.Tensor"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"_flatten_tensor","ret_type":{".class":"TupleType","implicit":false,"items":["torch._tensor.Tensor",{".class":"UnionType","items":["torch.distributed._tensor.placement_types.DTensorSpec",{".class":"NoneType"}]}],"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,"unpack_kwargs":false,"variables":[]}}},"_unflatten_tensor":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0,0],"arg_names":["tensor","spec"],"dataclass_transform_spec":null,"flags":[],"fullname":"torch.distributed.tensor.parallel._data_parallel_utils._unflatten_tensor","name":"_unflatten_tensor","type":{".class":"CallableType","arg_kinds":[0,0],"arg_names":["tensor","spec"],"arg_types":["torch._tensor.Tensor","torch.distributed._tensor.placement_types.DTensorSpec"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"_unflatten_tensor","ret_type":"torch._tensor.Tensor","type_guard":null,"unpack_kwargs":false,"variables":[]}}},"torch":{".class":"SymbolTableNode","cross_ref":"torch","kind":"Gdef"}},"path":"/home/bigcatc/miniconda3/envs/py310/lib/python3.10/site-packages/torch/distributed/tensor/parallel/_data_parallel_utils.py"}