icefall/.mypy_cache/3.9/torch/onnx/utils.data.json

{".class": "MypyFile", "_fullname": "torch.onnx.utils", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "ExportTypes": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.ExportTypes", "kind": "Gdef"}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "ListType": {".class": "SymbolTableNode", "cross_ref": "torch._C.ListType", "kind": "Gdef"}, "ONNXCheckerError": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "TypeInfo", "_promote": null, "abstract_attributes": [], "bases": ["builtins.Exception"], "declared_metaclass": null, "defn": {".class": "ClassDef", "fullname": "torch.onnx.utils.ONNXCheckerError", "name": "ONNXCheckerError", "type_vars": []}, "flags": [], "fullname": "torch.onnx.utils.ONNXCheckerError", "metaclass_type": null, "metadata": {}, "module_name": "torch.onnx.utils", "mro": ["torch.onnx.utils.ONNXCheckerError", "builtins.Exception", "builtins.BaseException", "builtins.object"], "names": {".class": "SymbolTable"}, "tuple_type": null, "type_vars": [], "typeddict_type": null}}, "ONNX_ARCHIVE_MODEL_PROTO_NAME": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.ONNX_ARCHIVE_MODEL_PROTO_NAME", "kind": "Gdef"}, "OperatorExportTypes": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.OperatorExportTypes", "kind": "Gdef"}, "OptionalType": {".class": "SymbolTableNode", "cross_ref": "torch._C.OptionalType", "kind": "Gdef"}, "TrainingMode": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.TrainingMode", "kind": "Gdef"}, "Tuple": {".class": "SymbolTableNode", "cross_ref": "typing.Tuple", "kind": "Gdef"}, "Union": {".class": "SymbolTableNode", "cross_ref": "typing.Union", "kind": "Gdef"}, "__IN_ONNX_EXPORT": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.onnx.utils.__IN_ONNX_EXPORT", "name": "__IN_ONNX_EXPORT", "type": "builtins.bool"}}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.utils.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.utils.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.utils.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.utils.__package__", "name": "__package__", "type": "builtins.str"}}, "_add_attribute": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["node", "key", "value", "aten"], "flags": [], "fullname": "torch.onnx.utils._add_attribute", "name": "_add_attribute", "type": null}}, "_add_block": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["node"], "flags": [], "fullname": "torch.onnx.utils._add_block", "name": "_add_block", "type": null}}, "_add_input_to_block": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["block"], "flags": [], "fullname": "torch.onnx.utils._add_input_to_block", "name": "_add_input_to_block", "type": null}}, "_add_output_to_block": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["block", "value"], "flags": [], "fullname": "torch.onnx.utils._add_output_to_block", "name": "_add_output_to_block", "type": null}}, "_block_op": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 2, 4], "arg_names": ["b", "opname", "args", "kwargs"], "flags": [], "fullname": "torch.onnx.utils._block_op", "name": "_block_op", "type": null}}, "_check_onnx_proto": {".class": "SymbolTableNode", "cross_ref": "torch._C._check_onnx_proto", "kind": "Gdef"}, "_create_jit_graph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["model", "args"], "flags": [], "fullname": "torch.onnx.utils._create_jit_graph", "name": "_create_jit_graph", "type": null}}, "_decide_add_node_names": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["add_node_names", "operator_export_type"], "flags": [], "fullname": "torch.onnx.utils._decide_add_node_names", "name": "_decide_add_node_names", "type": null}}, "_decide_constant_folding": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["do_constant_folding", "operator_export_type", "training"], "flags": [], "fullname": "torch.onnx.utils._decide_constant_folding", "name": "_decide_constant_folding", "type": null}}, "_decide_external_data_format": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["use_external_data_format", "operator_export_type", "f"], "flags": [], "fullname": "torch.onnx.utils._decide_external_data_format", "name": "_decide_external_data_format", "type": null}}, "_decide_input_format": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["model", "args"], "flags": [], "fullname": "torch.onnx.utils._decide_input_format", "name": "_decide_input_format", "type": null}}, "_decide_keep_init_as_input": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["keep_initializers_as_inputs", "operator_export_type", "opset_version"], "flags": [], "fullname": "torch.onnx.utils._decide_keep_init_as_input", "name": "_decide_keep_init_as_input", "type": null}}, "_export": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "f", "export_params", "verbose", "training", "input_names", "output_names", "operator_export_type", "export_type", "example_outputs", "opset_version", "do_constant_folding", "dynamic_axes", "keep_initializers_as_inputs", "fixed_batch_size", "custom_opsets", "add_node_names", "use_external_data_format", "onnx_shape_inference"], "flags": [], "fullname": "torch.onnx.utils._export", "name": "_export", "type": null}}, "_export_to_pretty_string": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "f", "export_params", "verbose", "training", "input_names", "output_names", "operator_export_type", "export_type", "example_outputs", "google_printer", "opset_version", "do_constant_folding", "keep_initializers_as_inputs", "fixed_batch_size", "custom_opsets", "add_node_names", "onnx_shape_inference", "dynamic_axes"], "flags": [], "fullname": "torch.onnx.utils._export_to_pretty_string", "name": "_export_to_pretty_string", "type": null}}, "_find_missing_ops_onnx_export": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "f", "verbose", "training", "input_names", "output_names", "opset_version", "dynamic_axes"], "flags": [], "fullname": "torch.onnx.utils._find_missing_ops_onnx_export", "name": "_find_missing_ops_onnx_export", "type": null}}, "_find_symbolic_in_registry": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["domain", "op_name", "opset_version", "operator_export_type"], "flags": [], "fullname": "torch.onnx.utils._find_symbolic_in_registry", "name": "_find_symbolic_in_registry", "type": null}}, "_get_example_outputs": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["model", "args"], "flags": [], "fullname": "torch.onnx.utils._get_example_outputs", "name": "_get_example_outputs", "type": null}}, "_get_named_param_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["graph", "params"], "flags": [], "fullname": "torch.onnx.utils._get_named_param_dict", "name": "_get_named_param_dict", "type": null}}, "_get_param_count_list": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["method_graph", "args_params"], "flags": [], "fullname": "torch.onnx.utils._get_param_count_list", "name": "_get_param_count_list", "type": null}}, "_graph_at": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 2, 4], "arg_names": ["g", "opname", "args", "kwargs"], "flags": [], "fullname": "torch.onnx.utils._graph_at", "name": "_graph_at", "type": null}}, "_graph_constant": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 2, 4], "arg_names": ["g", "value", "dims", "type", "args", "kwargs"], "flags": [], "fullname": "torch.onnx.utils._graph_constant", "name": "_graph_constant", "type": null}}, "_graph_op": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 2, 4], "arg_names": ["g", "opname", "raw_args", "kwargs"], "flags": [], "fullname": "torch.onnx.utils._graph_op", "name": "_graph_op", "type": null}}, "_is_constant_tensor_list": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["node"], "flags": [], "fullname": "torch.onnx.utils._is_constant_tensor_list", "name": "_is_constant_tensor_list", "type": null}}, "_is_onnx_list": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["value"], "flags": [], "fullname": "torch.onnx.utils._is_onnx_list", "name": "_is_onnx_list", "type": null}}, "_model_to_graph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "verbose", "input_names", "output_names", "operator_export_type", "example_outputs", "do_constant_folding", "_disable_torch_constant_prop", "fixed_batch_size", "training", "dynamic_axes"], "flags": [], "fullname": "torch.onnx.utils._model_to_graph", "name": "_model_to_graph", "type": null}}, "_newNode": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 2, 4], "arg_names": ["g", "opname", "outputs", "args", "kwargs"], "flags": [], "fullname": "torch.onnx.utils._newNode", "name": "_newNode", "type": null}}, "_node_getitem": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["self", "k"], "flags": [], "fullname": "torch.onnx.utils._node_getitem", "name": "_node_getitem", "type": null}}, "_optimize_graph": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1, 1], "arg_names": ["graph", "operator_export_type", "_disable_torch_constant_prop", "fixed_batch_size", "params_dict", "dynamic_axes", "input_names", "module"], "flags": [], "fullname": "torch.onnx.utils._optimize_graph", "name": "_optimize_graph", "type": null}}, "_params_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.onnx.utils._params_dict", "name": "_params_dict", "type": {".class": "Instance", "args": [{".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}], "type_ref": "builtins.dict"}}}, "_propagate_and_assign_input_shapes": {".class": "SymbolTableNode", "cross_ref": "torch._C._propagate_and_assign_input_shapes", "kind": "Gdef"}, "_resolve_args_by_export_type": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["arg_name", "arg_value", "operator_export_type"], "flags": [], "fullname": "torch.onnx.utils._resolve_args_by_export_type", "name": "_resolve_args_by_export_type", "type": null}}, "_run_symbolic_function": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 1], "arg_names": ["g", "block", "n", "inputs", "env", "operator_export_type"], "flags": [], "fullname": "torch.onnx.utils._run_symbolic_function", "name": "_run_symbolic_function", "type": null}}, "_run_symbolic_method": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["g", "op_name", "symbolic_fn", "args"], "flags": [], "fullname": "torch.onnx.utils._run_symbolic_method", "name": "_run_symbolic_method", "type": null}}, "_scalar": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["x"], "flags": [], "fullname": "torch.onnx.utils._scalar", "name": "_scalar", "type": null}}, "_set_input_and_output_names": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["graph", "input_names", "output_names"], "flags": [], "fullname": "torch.onnx.utils._set_input_and_output_names", "name": "_set_input_and_output_names", "type": null}}, "_split_tensor_list_constants": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "block"], "flags": [], "fullname": "torch.onnx.utils._split_tensor_list_constants", "name": "_split_tensor_list_constants", "type": null}}, "_trace": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["func", "args", "operator_export_type", "return_outs"], "flags": [], "fullname": "torch.onnx.utils._trace", "name": "_trace", "type": null}}, "_trace_and_get_graph_from_model": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["model", "args"], "flags": [], "fullname": "torch.onnx.utils._trace_and_get_graph_from_model", "name": "_trace_and_get_graph_from_model", "type": null}}, "_unique_state_dict": {".class": "SymbolTableNode", "cross_ref": "torch.jit._trace._unique_state_dict", "kind": "Gdef"}, "_validate_dynamic_axes": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0], "arg_names": ["dynamic_axes", "model", "input_names", "output_names"], "flags": [], "fullname": "torch.onnx.utils._validate_dynamic_axes", "name": "_validate_dynamic_axes", "type": null}}, "attr_pattern": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": [], "fullname": "torch.onnx.utils.attr_pattern", "name": "attr_pattern", "type": {".class": "Instance", "args": ["builtins.str"], "type_ref": "typing.Pattern"}}}, "collections": {".class": "SymbolTableNode", "cross_ref": "collections", "kind": "Gdef"}, "contextlib": {".class": "SymbolTableNode", "cross_ref": "contextlib", "kind": "Gdef"}, "copy": {".class": "SymbolTableNode", "cross_ref": "copy", "kind": "Gdef"}, "export": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "f", "export_params", "verbose", "training", "input_names", "output_names", "operator_export_type", "opset_version", "_retain_param_name", "do_constant_folding", "example_outputs", "strip_doc_string", "dynamic_axes", "keep_initializers_as_inputs", "custom_opsets", "enable_onnx_checker", "use_external_data_format"], "flags": [], "fullname": "torch.onnx.utils.export", "name": "export", "type": null}}, "export_to_pretty_string": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["model", "args", "f", "export_params", "verbose", "training", "input_names", "output_names", "operator_export_type", "export_type", "example_outputs", "google_printer", "opset_version", "_retain_param_name", "keep_initializers_as_inputs", "custom_opsets", "add_node_names", "do_constant_folding", "dynamic_axes"], "flags": [], "fullname": "torch.onnx.utils.export_to_pretty_string", "name": "export_to_pretty_string", "type": null}}, "get_ns_op_name_from_custom_op": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["symbolic_name"], "flags": [], "fullname": "torch.onnx.utils.get_ns_op_name_from_custom_op", "name": "get_ns_op_name_from_custom_op", "type": null}}, "is_in_onnx_export": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [], "arg_names": [], "flags": [], "fullname": "torch.onnx.utils.is_in_onnx_export", "name": "is_in_onnx_export", "type": null}}, "numbers": {".class": "SymbolTableNode", "cross_ref": "numbers", "kind": "Gdef"}, "re": {".class": "SymbolTableNode", "cross_ref": "re", "kind": "Gdef"}, "register_custom_op_symbolic": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["symbolic_name", "symbolic_fn", "opset_version"], "flags": [], "fullname": "torch.onnx.utils.register_custom_op_symbolic", "name": "register_custom_op_symbolic", "type": null}}, "select_model_mode_for_export": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["model", "mode"], "flags": ["is_generator", "is_decorated"], "fullname": "torch.onnx.utils.select_model_mode_for_export", "name": "select_model_mode_for_export", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "select_model_mode_for_export", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["model", "mode"], "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": false, "is_ellipsis_args": false, "name": "select_model_mode_for_export", "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": []}}}}, "string_classes": {".class": "SymbolTableNode", "cross_ref": "torch._six.string_classes", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "unregister_custom_op_symbolic": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["symbolic_name", "opset_version"], "flags": [], "fullname": "torch.onnx.utils.unregister_custom_op_symbolic", "name": "unregister_custom_op_symbolic", "type": null}}, "warn_on_static_input_change": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input_states"], "flags": [], "fullname": "torch.onnx.utils.warn_on_static_input_change", "name": "warn_on_static_input_change", "type": null}}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/onnx/utils.py"}