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

{".class": "MypyFile", "_fullname": "torch.onnx.symbolic_caffe2", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.symbolic_caffe2.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.symbolic_caffe2.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.symbolic_caffe2.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.onnx.symbolic_caffe2.__package__", "name": "__package__", "type": "builtins.str"}}, "_empty_affine_quantized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "shape", "scale", "zero_point", "dtype", "pin_memory", "memory_format", "layout"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2._empty_affine_quantized", "name": "_empty_affine_quantized", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "_empty_affine_quantized", "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}}}}, "_permute_helper": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["g", "input", "axes"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2._permute_helper", "name": "_permute_helper", "type": null}}, "add": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0], "arg_names": ["g", "input_a", "input_b", "scale", "zero_point"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.add", "name": "add", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "add", "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}}}}, "avg_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 1], "arg_names": ["g", "input", "kernel_size", "stride", "padding", "ceil_mode", "count_include_pad", "divisor_override"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.avg_pool2d", "name": "avg_pool2d", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "avg_pool2d", "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}}}}, "cat": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1], "arg_names": ["g", "tensor_list", "dim", "scale", "zero_point"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.cat", "name": "cat", "type": null}}, "conv2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "weight", "bias", "stride", "padding", "dilation", "groups", "scale", "zero_point"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.conv2d", "name": "conv2d", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "conv2d", "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}}}}, "conv2d_relu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "weight", "bias", "stride", "padding", "dilation", "groups", "scale", "zero_point"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.conv2d_relu", "name": "conv2d_relu", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "conv2d_relu", "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}}}}, "conv_prepack": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "weight", "bias", "stride", "padding", "dilation", "groups"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.conv_prepack", "name": "conv_prepack", "type": null}}, "dequantize": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "input"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.dequantize", "name": "dequantize", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "dequantize", "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}}}}, "getmembers": {".class": "SymbolTableNode", "cross_ref": "inspect.getmembers", "kind": "Gdef"}, "importlib": {".class": "SymbolTableNode", "cross_ref": "importlib", "kind": "Gdef"}, "isfunction": {".class": "SymbolTableNode", "cross_ref": "inspect.isfunction", "kind": "Gdef"}, "linear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "weight", "bias", "scale", "zero_point"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.linear", "name": "linear", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "linear", "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}}}}, "linear_prepack": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["g", "weight", "bias"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.linear_prepack", "name": "linear_prepack", "type": null}}, "max_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "kernel_size", "stride", "padding", "dilation", "ceil_mode"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.max_pool2d", "name": "max_pool2d", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "max_pool2d", "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}}}}, "nchw2nhwc": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "input"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.nchw2nhwc", "name": "nchw2nhwc", "type": null}}, "nhwc2nchw": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "input"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.nhwc2nchw", "name": "nhwc2nchw", "type": null}}, "parse_args": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.symbolic_helper.parse_args", "kind": "Gdef"}, "quantize_per_tensor": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0], "arg_names": ["g", "input", "scale", "zero_point", "dtype"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.quantize_per_tensor", "name": "quantize_per_tensor", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "quantize_per_tensor", "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}}}}, "register_quantized_ops": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["domain", "version"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.register_quantized_ops", "name": "register_quantized_ops", "type": null}}, "relu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "input"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.relu", "name": "relu", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "relu", "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}}}}, "reshape": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["g", "input", "shape"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.reshape", "name": "reshape", "type": null}}, "sigmoid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["g", "input"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.sigmoid", "name": "sigmoid", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "sigmoid", "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}}}}, "slice": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Decorator", "func": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0, 0], "arg_names": ["g", "input", "dim", "start", "end", "step"], "flags": ["is_decorated"], "fullname": "torch.onnx.symbolic_caffe2.slice", "name": "slice", "type": null}, "is_overload": false, "var": {".class": "Var", "flags": ["is_ready"], "fullname": null, "name": "slice", "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}}}}, "sym_help": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.symbolic_helper", "kind": "Gdef"}, "sym_registry": {".class": "SymbolTableNode", "cross_ref": "torch.onnx.symbolic_registry", "kind": "Gdef"}, "upsample_nearest2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1], "arg_names": ["g", "input", "output_size", "align_corners", "scales_h", "scales_w"], "flags": [], "fullname": "torch.onnx.symbolic_caffe2.upsample_nearest2d", "name": "upsample_nearest2d", "type": null}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/onnx/symbolic_caffe2.py"}