icefall/.mypy_cache/3.9/torch/autograd/function.data.json

{".class": "MypyFile", "_fullname": "torch.autograd.function", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef"}, "BackwardCFunction": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch._C._FunctionBase", "torch.autograd.function.FunctionCtx", "torch.autograd.function._HookMixin"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.BackwardCFunction", "name": "BackwardCFunction", "type_vars": []}, "flags": [], "fullname": "torch.autograd.function.BackwardCFunction", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.BackwardCFunction", "torch._C._FunctionBase", "torch.autograd.function.FunctionCtx", "torch.autograd.function._HookMixin", "builtins.object"], "names": {".class": "SymbolTable", "apply": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.BackwardCFunction.apply", "name": "apply", "type": null}}, "apply_jvp": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.BackwardCFunction.apply_jvp", "name": "apply_jvp", "type": null}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Function": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.object"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.Function", "name": "Function", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.autograd.function.Function", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.Function", "builtins.object"], "names": {".class": "SymbolTable", "__call__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.autograd.function.Function.__call__", "name": "__call__", "type": null}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.autograd.function.Function.__init__", "name": "__init__", "type": null}}, "backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_outputs"], "flags": ["is_static", "is_decorated"], "fullname": "torch.autograd.function.Function.backward", "name": "backward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_outputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "backward of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "backward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_outputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "backward of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["ctx", "args", "kwargs"], "flags": ["is_static", "is_decorated"], "fullname": "torch.autograd.function.Function.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 2, 4], "arg_names": ["ctx", "args", "kwargs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 2, 4], "arg_names": ["ctx", "args", "kwargs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}}, "is_traceable": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.autograd.function.Function.is_traceable", "name": "is_traceable", "type": "builtins.bool"}}, "jvp": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_inputs"], "flags": ["is_static", "is_decorated"], "fullname": "torch.autograd.function.Function.jvp", "name": "jvp", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_inputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "jvp of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "jvp", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_inputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "jvp of Function", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}}, "vjp": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.autograd.function.Function.vjp", "name": "vjp", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["ctx", "grad_outputs"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "FunctionCtx": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.object"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.FunctionCtx", "name": "FunctionCtx", "type_vars": []}, "flags": [], "fullname": "torch.autograd.function.FunctionCtx", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.FunctionCtx", "builtins.object"], "names": {".class": "SymbolTable", "dirty_tensors": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.FunctionCtx.dirty_tensors", "name": "dirty_tensors", "type": {".class": "Instance", "args": ["torch._tensor.Tensor"], "type_ref": "builtins.tuple"}}}, "mark_dirty": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.FunctionCtx.mark_dirty", "name": "mark_dirty", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "arg_types": ["torch.autograd.function.FunctionCtx", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mark_dirty of FunctionCtx", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "mark_non_differentiable": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.FunctionCtx.mark_non_differentiable", "name": "mark_non_differentiable", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "arg_types": ["torch.autograd.function.FunctionCtx", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mark_non_differentiable of FunctionCtx", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "mark_shared_storage": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "pairs"], "flags": [], "fullname": "torch.autograd.function.FunctionCtx.mark_shared_storage", "name": "mark_shared_storage", "type": null}}, "materialize_grads": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.FunctionCtx.materialize_grads", "name": "materialize_grads", "type": "builtins.bool"}}, "non_differentiable": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.FunctionCtx.non_differentiable", "name": "non_differentiable", "type": {".class": "Instance", "args": ["torch._tensor.Tensor"], "type_ref": "builtins.tuple"}}}, "save_for_backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "tensors"], "flags": [], "fullname": "torch.autograd.function.FunctionCtx.save_for_backward", "name": "save_for_backward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "tensors"], "arg_types": ["torch.autograd.function.FunctionCtx", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "save_for_backward of FunctionCtx", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "set_materialize_grads": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "flags": [], "fullname": "torch.autograd.function.FunctionCtx.set_materialize_grads", "name": "set_materialize_grads", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["self", "value"], "arg_types": ["torch.autograd.function.FunctionCtx", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "set_materialize_grads of FunctionCtx", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "to_save": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.FunctionCtx.to_save", "name": "to_save", "type": {".class": "Instance", "args": ["torch._tensor.Tensor"], "type_ref": "builtins.tuple"}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "FunctionMeta": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.type"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.FunctionMeta", "name": "FunctionMeta", "type_vars": []}, "flags": [], "fullname": "torch.autograd.function.FunctionMeta", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.FunctionMeta", "builtins.type", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["cls", "name", "bases", "attrs"], "flags": [], "fullname": "torch.autograd.function.FunctionMeta.__init__", "name": "__init__", "type": null}}, "_backward_cls": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.FunctionMeta._backward_cls", "name": "_backward_cls", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "InplaceFunction": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.autograd.function.Function"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.InplaceFunction", "name": "InplaceFunction", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.autograd.function.InplaceFunction", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.InplaceFunction", "torch.autograd.function.Function", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["self", "inplace"], "flags": [], "fullname": "torch.autograd.function.InplaceFunction.__init__", "name": "__init__", "type": null}}, "inplace": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.InplaceFunction.inplace", "name": "inplace", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "NestedIOFunction": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.autograd.function.Function"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function.NestedIOFunction", "name": "NestedIOFunction", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.autograd.function.NestedIOFunction", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function.NestedIOFunction", "torch.autograd.function.Function", "builtins.object"], "names": {".class": "SymbolTable", "__call__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.autograd.function.NestedIOFunction.__call__", "name": "__call__", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "input"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": null, "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "_do_backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "gradients", "retain_variables"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction._do_backward", "name": "_do_backward", "type": null}}, "_do_forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction._do_forward", "name": "_do_forward", "type": null}}, "_nested_input": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction._nested_input", "name": "_nested_input", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_nested_output": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction._nested_output", "name": "_nested_output", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 5}}}, "_to_save_nested": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction._to_save_nested", "name": "_to_save_nested", "type": {".class": "Instance", "args": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "type_ref": "builtins.tuple"}}}, "backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "gradients"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.backward", "name": "backward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "gradients"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "backward of NestedIOFunction", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}, "backward_extended": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "grad_output"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.backward_extended", "name": "backward_extended", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "grad_output"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "backward_extended of NestedIOFunction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "dirty_tensors": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.dirty_tensors", "name": "dirty_tensors", "type": {".class": "Instance", "args": [{".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}], "type_ref": "builtins.tuple"}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of NestedIOFunction", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}, "forward_extended": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "input"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.forward_extended", "name": "forward_extended", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "input"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward_extended of NestedIOFunction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "mark_dirty": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.mark_dirty", "name": "mark_dirty", "type": {".class": "CallableType", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mark_dirty of NestedIOFunction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "mark_non_differentiable": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.mark_non_differentiable", "name": "mark_non_differentiable", "type": {".class": "CallableType", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "mark_non_differentiable of NestedIOFunction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "non_differentiable": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.non_differentiable", "name": "non_differentiable", "type": {".class": "Instance", "args": [{".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}], "type_ref": "builtins.tuple"}}}, "retain_variables": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.retain_variables", "name": "retain_variables", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "save_for_backward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.save_for_backward", "name": "save_for_backward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "arg_types": ["torch.autograd.function.NestedIOFunction", {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "save_for_backward of NestedIOFunction", "ret_type": {".class": "NoneType"}, "type_guard": null, "variables": []}}}, "saved_tensors": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": ["is_property", "is_decorated"], "fullname": "torch.autograd.function.NestedIOFunction.saved_tensors", "name": "saved_tensors", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_property", "is_ready"], "fullname": null, "name": "saved_tensors", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["self"], "arg_types": ["torch.autograd.function.NestedIOFunction"], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": "saved_tensors of NestedIOFunction", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}, "to_save": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function.NestedIOFunction.to_save", "name": "to_save", "type": {".class": "Instance", "args": [{".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}], "type_ref": "builtins.tuple"}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "OrderedDict": {".class": "SymbolTableNode", "cross_ref": "collections.OrderedDict", "kind": "Gdef"}, "_C": {".class": "SymbolTableNode", "cross_ref": "torch._C", "kind": "Gdef"}, "_ContextMethodMixin": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeAlias", "alias_tvars": [], "column": 0, "fullname": "torch.autograd.function._ContextMethodMixin", "line": 175, "no_args": true, "normalized": false, "target": "torch.autograd.function.FunctionCtx"}}, "_HookMixin": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.object"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.autograd.function._HookMixin", "name": "_HookMixin", "type_vars": []}, "flags": [], "fullname": "torch.autograd.function._HookMixin", "metaclass_type": null, "metadata": {}, "module_name": "torch.autograd.function", "mro": ["torch.autograd.function._HookMixin", "builtins.object"], "names": {".class": "SymbolTable", "_register_hook": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["backward_hooks", "hook"], "flags": ["is_static", "is_decorated"], "fullname": "torch.autograd.function._HookMixin._register_hook", "name": "_register_hook", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "_register_hook", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["backward_hooks", "hook"], "arg_types": [{".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": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": "_register_hook of _HookMixin", "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}}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.autograd.function.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.autograd.function.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.autograd.function.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.autograd.function.__package__", "name": "__package__", "type": "builtins.str"}}, "_functions": {".class": "SymbolTableNode", "cross_ref": "torch._C._functions", "kind": "Gdef"}, "_iter_None_tensors": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function._iter_None_tensors", "name": "_iter_None_tensors", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_iter_filter": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["condition", "allow_unknown", "condition_msg", "conversion"], "flags": [], "fullname": "torch.autograd.function._iter_filter", "name": "_iter_filter", "type": null}}, "_iter_jit_values": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function._iter_jit_values", "name": "_iter_jit_values", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_iter_tensors": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function._iter_tensors", "name": "_iter_tensors", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_iter_tensors_permissive": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function._iter_tensors_permissive", "name": "_iter_tensors_permissive", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_jit_unwrap_structured": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["obj"], "flags": [], "fullname": "torch.autograd.function._jit_unwrap_structured", "name": "_jit_unwrap_structured", "type": null}}, "_map_tensor_data": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.autograd.function._map_tensor_data", "name": "_map_tensor_data", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_nested_map": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["condition", "fn", "condition_msg"], "flags": [], "fullname": "torch.autograd.function._nested_map", "name": "_nested_map", "type": null}}, "_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "proto"], "flags": [], "fullname": "torch.autograd.function._unflatten", "name": "_unflatten", "type": null}}, "functools": {".class": "SymbolTableNode", "cross_ref": "functools", "kind": "Gdef"}, "hooks": {".class": "SymbolTableNode", "cross_ref": "torch.utils.hooks", "kind": "Gdef"}, "once_differentiable": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["fn"], "flags": [], "fullname": "torch.autograd.function.once_differentiable", "name": "once_differentiable", "type": null}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "traceable": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["fn_cls"], "flags": [], "fullname": "torch.autograd.function.traceable", "name": "traceable", "type": null}}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}, "with_metaclass": {".class": "SymbolTableNode", "cross_ref": "torch._six.with_metaclass", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/autograd/function.py"}