{".class":"MypyFile","_fullname":"torch._functorch.batch_norm_replacement","future_import_flags":[],"is_partial_stub_package":false,"is_stub":false,"names":{".class":"SymbolTable","__annotations__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._functorch.batch_norm_replacement.__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._functorch.batch_norm_replacement.__doc__","name":"__doc__","type":"builtins.str"}},"__file__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._functorch.batch_norm_replacement.__file__","name":"__file__","type":"builtins.str"}},"__name__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._functorch.batch_norm_replacement.__name__","name":"__name__","type":"builtins.str"}},"__package__":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Var","flags":["is_ready"],"fullname":"torch._functorch.batch_norm_replacement.__package__","name":"__package__","type":"builtins.str"}},"batch_norm_without_running_stats":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["module"],"dataclass_transform_spec":null,"flags":[],"fullname":"torch._functorch.batch_norm_replacement.batch_norm_without_running_stats","name":"batch_norm_without_running_stats","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["module"],"arg_types":["torch.nn.modules.module.Module"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"batch_norm_without_running_stats","ret_type":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_guard":null,"unpack_kwargs":false,"variables":[]}}},"exposed_in":{".class":"SymbolTableNode","cross_ref":"torch._functorch.utils.exposed_in","kind":"Gdef"},"nn":{".class":"SymbolTableNode","cross_ref":"torch.nn","kind":"Gdef"},"replace_all_batch_norm_modules_":{".class":"SymbolTableNode","kind":"Gdef","node":{".class":"Decorator","func":{".class":"FuncDef","abstract_status":0,"arg_kinds":[0],"arg_names":["root"],"dataclass_transform_spec":null,"flags":["is_decorated"],"fullname":"torch._functorch.batch_norm_replacement.replace_all_batch_norm_modules_","name":"replace_all_batch_norm_modules_","type":{".class":"CallableType","arg_kinds":[0],"arg_names":["root"],"arg_types":["torch.nn.modules.module.Module"],"bound_args":[],"def_extras":{"first_arg":null},"fallback":"builtins.function","from_concatenate":false,"implicit":false,"imprecise_arg_kinds":false,"is_ellipsis_args":false,"name":"replace_all_batch_norm_modules_","ret_type":"torch.nn.modules.module.Module","type_guard":null,"unpack_kwargs":false,"variables":[]}},"is_overload":false,"var":{".class":"Var","flags":["is_ready","is_inferred"],"fullname":"torch._functorch.batch_norm_replacement.replace_all_batch_norm_modules_","name":"replace_all_batch_norm_modules_","type":{".class":"AnyType","missing_import_name":null,"source_any":{".class":"AnyType","missing_import_name":null,"source_any":null,"type_of_any":1},"type_of_any":7}}}}},"path":"/home/bigcatc/miniconda3/envs/py310/lib/python3.10/site-packages/torch/_functorch/batch_norm_replacement.py"}