mirrors/icefall
1
0
Fork 0
mirror of https://github.com/k2-fsa/icefall.git synced 2025-09-05 07:04:18 +00:00
Code Issues Packages Projects Releases Wiki Activity
icefall/.mypy_cache/3.9/torch/jit/_trace.data.json
Patrick von Platen 0c7fe37e2f add hf hub
2021-12-08 23:01:51 +01:00

1 line
23 KiB
JSON
Raw Blame History

{".class": "MypyFile", "_fullname": "torch.jit._trace", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef"}, "Dict": {".class": "SymbolTableNode", "cross_ref": "typing.Dict", "kind": "Gdef"}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "Module": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.module.Module", "kind": "Gdef"}, "ONNXTracedModule": {".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.jit._trace.ONNXTracedModule", "name": "ONNXTracedModule", "type_vars": []}, "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.ONNXTracedModule", "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], "arg_names": ["self", "inner", "strict", "force_outplace", "return_inputs", "return_inputs_states"], "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule.__init__", "name": "__init__", "type": null}}, "_force_outplace": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule._force_outplace", "name": "_force_outplace", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_return_inputs": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule._return_inputs", "name": "_return_inputs", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "_return_inputs_states": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule._return_inputs_states", "name": "_return_inputs_states", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule.forward", "name": "forward", "type": {".class": "CallableType", "arg_kinds": [0, 2], "arg_names": ["self", "args"], "arg_types": ["torch.jit._trace.ONNXTracedModule", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": "self"}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "forward of ONNXTracedModule", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "inner": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule.inner", "name": "inner", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}, "strict": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.ONNXTracedModule.strict", "name": "strict", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "QualnameWrapper@1025": {".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.jit._trace.QualnameWrapper@1025", "name": "QualnameWrapper", "type_vars": []}, "flags": [], "fullname": "torch.jit._trace.QualnameWrapper@1025", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.QualnameWrapper@1025", "torch.nn.modules.module.Module", "builtins.object"], "names": {".class": "SymbolTable"}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "ScriptModule": {".class": "SymbolTableNode", "cross_ref": "torch.jit._script.ScriptModule", "kind": "Gdef"}, "Set": {".class": "SymbolTableNode", "cross_ref": "typing.Set", "kind": "Gdef"}, "TopLevelTracedModule": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.jit._trace.TracedModule"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.jit._trace.TopLevelTracedModule", "name": "TopLevelTracedModule", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.jit._trace.TopLevelTracedModule", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.TopLevelTracedModule", "torch.jit._trace.TracedModule", "torch.jit._script.ScriptModule", "builtins.object"], "names": {".class": "SymbolTable", "_reconstruct": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "cpp_module"], "flags": [], "fullname": "torch.jit._trace.TopLevelTracedModule._reconstruct", "name": "_reconstruct", "type": null}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.jit._trace.TopLevelTracedModule.forward", "name": "forward", "type": "torch.jit._script._CachedForward"}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "TracedModule": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["torch.jit._script.ScriptModule"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.jit._trace.TracedModule", "name": "TracedModule", "type_vars": []}, "flags": ["fallback_to_any"], "fullname": "torch.jit._trace.TracedModule", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.TracedModule", "torch.jit._script.ScriptModule", "builtins.object"], "names": {".class": "SymbolTable", "__getattr__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "attr"], "flags": [], "fullname": "torch.jit._trace.TracedModule.__getattr__", "name": "__getattr__", "type": null}}, "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["self", "orig", "id_set", "_compilation_unit"], "flags": [], "fullname": "torch.jit._trace.TracedModule.__init__", "name": "__init__", "type": null}}, "__setattr__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["self", "attr", "value"], "flags": [], "fullname": "torch.jit._trace.TracedModule.__setattr__", "name": "__setattr__", "type": null}}, "_disable_script_meta": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Var", "flags": ["is_initialized_in_class"], "fullname": "torch.jit._trace.TracedModule._disable_script_meta", "name": "_disable_script_meta", "type": "builtins.bool"}}, "_get_name": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.jit._trace.TracedModule._get_name", "name": "_get_name", "type": null}}, "extra_repr": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["self"], "flags": [], "fullname": "torch.jit._trace.TracedModule.extra_repr", "name": "extra_repr", "type": null}}, "forward": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 2, 4], "arg_names": ["self", "args", "kwargs"], "flags": [], "fullname": "torch.jit._trace.TracedModule.forward", "name": "forward", "type": null}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "TracerWarning": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.Warning"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.jit._trace.TracerWarning", "name": "TracerWarning", "type_vars": []}, "flags": [], "fullname": "torch.jit._trace.TracerWarning", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.TracerWarning", "builtins.Warning", "builtins.Exception", "builtins.BaseException", "builtins.object"], "names": {".class": "SymbolTable", "ignore_lib_warnings": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": ["is_static", "is_decorated"], "fullname": "torch.jit._trace.TracerWarning.ignore_lib_warnings", "name": "ignore_lib_warnings", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_initialized_in_class", "is_staticmethod", "is_ready"], "fullname": null, "name": "ignore_lib_warnings", "type": {".class": "CallableType", "arg_kinds": [], "arg_names": [], "arg_types": [], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": true, "is_ellipsis_args": false, "name": "ignore_lib_warnings of TracerWarning", "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}}, "TracingCheckError": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.Exception"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.jit._trace.TracingCheckError", "name": "TracingCheckError", "type_vars": []}, "flags": [], "fullname": "torch.jit._trace.TracingCheckError", "metaclass_type": null, "metadata": {}, "module_name": "torch.jit._trace", "mro": ["torch.jit._trace.TracingCheckError", "builtins.Exception", "builtins.BaseException", "builtins.object"], "names": {".class": "SymbolTable", "__init__": {".class": "SymbolTableNode", "kind": "Mdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["self", "graph_diff_error", "tensor_compare_error", "extra_msg"], "flags": [], "fullname": "torch.jit._trace.TracingCheckError.__init__", "name": "__init__", "type": null}}, "message": {".class": "SymbolTableNode", "implicit": true, "kind": "Mdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace.TracingCheckError.message", "name": "message", "type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}}}}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "_CachedForward": {".class": "SymbolTableNode", "cross_ref": "torch.jit._script._CachedForward", "kind": "Gdef"}, "_JIT_DISABLE": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace._JIT_DISABLE", "name": "_JIT_DISABLE", "type": {".class": "UnionType", "items": ["builtins.str", "builtins.bool"]}}}, "_JIT_STATS": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace._JIT_STATS", "name": "_JIT_STATS", "type": {".class": "UnionType", "items": ["builtins.str", "builtins.bool"]}}}, "_JIT_TIME": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace._JIT_TIME", "name": "_JIT_TIME", "type": {".class": "UnionType", "items": ["builtins.str", "builtins.bool"]}}}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.jit._trace.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.jit._trace.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.jit._trace.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.jit._trace.__package__", "name": "__package__", "type": "builtins.str"}}, "_check_trace": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["check_inputs", "func", "traced_func", "check_tolerance", "strict", "force_outplace", "is_trace_module", "_module_class"], "flags": ["is_decorated"], "fullname": "torch.jit._trace._check_trace", "name": "_check_trace", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "_check_trace", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["check_inputs", "func", "traced_func", "check_tolerance", "strict", "force_outplace", "is_trace_module", "_module_class"], "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}, {".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}, {".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}, {".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": "_check_trace", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}}, "_clone_inputs": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["args"], "flags": [], "fullname": "torch.jit._trace._clone_inputs", "name": "_clone_inputs", "type": null}}, "_create_interpreter_name_lookup_fn": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [1], "arg_names": ["frames_up"], "flags": [], "fullname": "torch.jit._trace._create_interpreter_name_lookup_fn", "name": "_create_interpreter_name_lookup_fn", "type": null}}, "_enabled": {".class": "SymbolTableNode", "cross_ref": "torch.jit._state._enabled", "kind": "Gdef"}, "_flatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace._flatten", "name": "_flatten", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["arg"], "arg_types": [{".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": "TupleType", "implicit": false, "items": [{".class": "Instance", "args": ["torch._tensor.Tensor"], "type_ref": "builtins.list"}, "torch._C.IODescriptor"], "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, "variables": []}}}, "_get_default_rtol_and_atol": {".class": "SymbolTableNode", "cross_ref": "torch.testing._asserts._get_default_rtol_and_atol", "kind": "Gdef"}, "_get_trace_graph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1, 1, 1], "arg_names": ["f", "args", "kwargs", "strict", "_force_outplace", "return_inputs", "_return_inputs_states"], "flags": [], "fullname": "torch.jit._trace._get_trace_graph", "name": "_get_trace_graph", "type": null}}, "_python_cu": {".class": "SymbolTableNode", "cross_ref": "torch.jit._state._python_cu", "kind": "Gdef"}, "_qualified_name": {".class": "SymbolTableNode", "cross_ref": "torch._jit_internal._qualified_name", "kind": "Gdef"}, "_script_if_tracing": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["fn"], "flags": [], "fullname": "torch.jit._trace._script_if_tracing", "name": "_script_if_tracing", "type": null}}, "_time": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 1], "arg_names": ["trace_name", "name", "time"], "flags": ["is_generator", "is_decorated"], "fullname": "torch.jit._trace._time", "name": "_time", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "_time", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1], "arg_names": ["trace_name", "name", "time"], "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}, {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}], "bound_args": [], "def_extras": {}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "_time", "ret_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": "contextlib._GeneratorContextManager"}, "type_guard": null, "variables": []}}}}, "_trace_module_map": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.jit._trace._trace_module_map", "name": "_trace_module_map", "type": {".class": "UnionType", "items": [{".class": "Instance", "args": [{".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}], "type_ref": "builtins.dict"}, {".class": "NoneType"}]}}}, "_unflatten": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.jit._trace._unflatten", "name": "_unflatten", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["vars", "desc"], "arg_types": [{".class": "Instance", "args": ["torch._tensor.Tensor"], "type_ref": "builtins.list"}, "torch._C.IODescriptor"], "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": []}}}, "_unique_state_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1], "arg_names": ["module", "keep_vars"], "flags": [], "fullname": "torch.jit._trace._unique_state_dict", "name": "_unique_state_dict", "type": null}}, "_verify_equal": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["xs", "ys"], "flags": [], "fullname": "torch.jit._trace._verify_equal", "name": "_verify_equal", "type": null}}, "contextlib": {".class": "SymbolTableNode", "cross_ref": "contextlib", "kind": "Gdef"}, "copy": {".class": "SymbolTableNode", "cross_ref": "copy", "kind": "Gdef"}, "function": {".class": "SymbolTableNode", "cross_ref": "torch.autograd.function", "kind": "Gdef"}, "functools": {".class": "SymbolTableNode", "cross_ref": "functools", "kind": "Gdef"}, "get_callable_argument_names": {".class": "SymbolTableNode", "cross_ref": "torch._jit_internal.get_callable_argument_names", "kind": "Gdef"}, "indent": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["s"], "flags": [], "fullname": "torch.jit._trace.indent", "name": "indent", "type": null}}, "inspect": {".class": "SymbolTableNode", "cross_ref": "inspect", "kind": "Gdef"}, "is_scripting": {".class": "SymbolTableNode", "cross_ref": "torch._jit_internal.is_scripting", "kind": "Gdef"}, "is_tracing": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.jit._trace.is_tracing", "name": "is_tracing", "type": null}}, "make_module": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["mod", "_module_class", "_compilation_unit"], "flags": [], "fullname": "torch.jit._trace.make_module", "name": "make_module", "type": null}}, "make_tuple": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["example_inputs"], "flags": [], "fullname": "torch.jit._trace.make_tuple", "name": "make_tuple", "type": null}}, "os": {".class": "SymbolTableNode", "cross_ref": "os", "kind": "Gdef"}, "re": {".class": "SymbolTableNode", "cross_ref": "re", "kind": "Gdef"}, "script": {".class": "SymbolTableNode", "cross_ref": "torch.jit._script.script", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "trace": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["func", "example_inputs", "optimize", "check_trace", "check_inputs", "check_tolerance", "strict", "_force_outplace", "_module_class", "_compilation_unit"], "flags": [], "fullname": "torch.jit._trace.trace", "name": "trace", "type": null}}, "trace_module": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["mod", "inputs", "optimize", "check_trace", "check_inputs", "check_tolerance", "strict", "_force_outplace", "_module_class", "_compilation_unit"], "flags": [], "fullname": "torch.jit._trace.trace_module", "name": "trace_module", "type": null}}, "verify": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1], "arg_names": ["model", "args", "loss_fn", "devices"], "flags": [], "fullname": "torch.jit._trace.verify", "name": "verify", "type": null}}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}, "wrap_check_inputs": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["check_inputs"], "flags": [], "fullname": "torch.jit._trace.wrap_check_inputs", "name": "wrap_check_inputs", "type": null}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/jit/_trace.py"}
Reference in New Issue View Git Blame Copy Permalink