icefall/.mypy_cache/3.9/torch/nn/quantized/functional.data.json

{".class": "MypyFile", "_fullname": "torch.nn.quantized.functional", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "BroadcastingList2": {".class": "SymbolTableNode", "cross_ref": "torch.jit.annotations.BroadcastingList2", "kind": "Gdef"}, "List": {".class": "SymbolTableNode", "cross_ref": "typing.List", "kind": "Gdef"}, "Optional": {".class": "SymbolTableNode", "cross_ref": "typing.Optional", "kind": "Gdef"}, "Tensor": {".class": "SymbolTableNode", "cross_ref": "torch._tensor.Tensor", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.quantized.functional.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.quantized.functional.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.quantized.functional.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.nn.quantized.functional.__package__", "name": "__package__", "type": "builtins.str"}}, "_pair": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.utils._pair", "kind": "Gdef"}, "_pair_from_first": {".class": "SymbolTableNode", "cross_ref": "torch.nn.quantized.modules.utils._pair_from_first", "kind": "Gdef"}, "_triple": {".class": "SymbolTableNode", "cross_ref": "torch.nn.modules.utils._triple", "kind": "Gdef"}, "adaptive_avg_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "flags": [], "fullname": "torch.nn.quantized.functional.adaptive_avg_pool2d", "name": "adaptive_avg_pool2d", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "arg_types": ["torch._tensor.Tensor", {".class": "AnyType", "missing_import_name": "torch.jit.annotations.BroadcastingList2", "source_any": null, "type_of_any": 3}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_avg_pool2d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "adaptive_avg_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "flags": [], "fullname": "torch.nn.quantized.functional.adaptive_avg_pool3d", "name": "adaptive_avg_pool3d", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["input", "output_size"], "arg_types": ["torch._tensor.Tensor", {".class": "AnyType", "missing_import_name": "torch.jit.annotations.BroadcastingList2", "source_any": null, "type_of_any": 3}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "adaptive_avg_pool3d", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "avg_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "ceil_mode", "count_include_pad", "divisor_override"], "flags": [], "fullname": "torch.nn.quantized.functional.avg_pool2d", "name": "avg_pool2d", "type": null}}, "avg_pool3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "ceil_mode", "count_include_pad", "divisor_override"], "flags": [], "fullname": "torch.nn.quantized.functional.avg_pool3d", "name": "avg_pool3d", "type": null}}, "celu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "scale", "zero_point", "alpha"], "flags": [], "fullname": "torch.nn.quantized.functional.celu", "name": "celu", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "scale", "zero_point", "alpha"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.int", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "celu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "clamp": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["input", "min_", "max_"], "flags": [], "fullname": "torch.nn.quantized.functional.clamp", "name": "clamp", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["input", "min_", "max_"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "clamp", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "conv1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "bias", "stride", "padding", "dilation", "groups", "padding_mode", "scale", "zero_point", "dtype"], "flags": [], "fullname": "torch.nn.quantized.functional.conv1d", "name": "conv1d", "type": null}}, "conv2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "bias", "stride", "padding", "dilation", "groups", "padding_mode", "scale", "zero_point", "dtype"], "flags": [], "fullname": "torch.nn.quantized.functional.conv2d", "name": "conv2d", "type": null}}, "conv3d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1, 1, 1, 1, 1, 1, 1, 1], "arg_names": ["input", "weight", "bias", "stride", "padding", "dilation", "groups", "padding_mode", "scale", "zero_point", "dtype"], "flags": [], "fullname": "torch.nn.quantized.functional.conv3d", "name": "conv3d", "type": null}}, "elu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "scale", "zero_point", "alpha"], "flags": [], "fullname": "torch.nn.quantized.functional.elu", "name": "elu", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 1], "arg_names": ["input", "scale", "zero_point", "alpha"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.int", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "elu", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardsigmoid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["input"], "flags": [], "fullname": "torch.nn.quantized.functional.hardsigmoid", "name": "hardsigmoid", "type": {".class": "CallableType", "arg_kinds": [0], "arg_names": ["input"], "arg_types": ["torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardsigmoid", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardswish": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["input", "scale", "zero_point"], "flags": [], "fullname": "torch.nn.quantized.functional.hardswish", "name": "hardswish", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["input", "scale", "zero_point"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.int"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardswish", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "hardtanh": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "min_val", "max_val", "inplace"], "flags": [], "fullname": "torch.nn.quantized.functional.hardtanh", "name": "hardtanh", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1], "arg_names": ["input", "min_val", "max_val", "inplace"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "hardtanh", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "interpolate": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "flags": [], "fullname": "torch.nn.quantized.functional.interpolate", "name": "interpolate", "type": null}}, "leaky_relu": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "negative_slope", "inplace", "scale", "zero_point"], "flags": [], "fullname": "torch.nn.quantized.functional.leaky_relu", "name": "leaky_relu", "type": {".class": "CallableType", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "negative_slope", "inplace", "scale", "zero_point"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.bool", {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "leaky_relu", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 1}, "type_guard": null, "variables": []}}}, "linear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "weight", "bias", "scale", "zero_point"], "flags": [], "fullname": "torch.nn.quantized.functional.linear", "name": "linear", "type": {".class": "CallableType", "arg_kinds": [0, 0, 1, 1, 1], "arg_names": ["input", "weight", "bias", "scale", "zero_point"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor", {".class": "UnionType", "items": ["torch._tensor.Tensor", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.float", {".class": "NoneType"}]}, {".class": "UnionType", "items": ["builtins.int", {".class": "NoneType"}]}], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "linear", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "max_pool1d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "flags": [], "fullname": "torch.nn.quantized.functional.max_pool1d", "name": "max_pool1d", "type": null}}, "max_pool2d": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 1, 1, 1, 1, 1], "arg_names": ["input", "kernel_size", "stride", "padding", "dilation", "ceil_mode", "return_indices"], "flags": [], "fullname": "torch.nn.quantized.functional.max_pool2d", "name": "max_pool2d", "type": null}}, "threshold": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["input", "threshold", "value"], "flags": [], "fullname": "torch.nn.quantized.functional.threshold", "name": "threshold", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0], "arg_names": ["input", "threshold", "value"], "arg_types": ["torch._tensor.Tensor", "builtins.float", "builtins.float"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "threshold", "ret_type": "torch._tensor.Tensor", "type_guard": null, "variables": []}}}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "upsample": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1, 1, 1], "arg_names": ["input", "size", "scale_factor", "mode", "align_corners"], "flags": [], "fullname": "torch.nn.quantized.functional.upsample", "name": "upsample", "type": null}}, "upsample_bilinear": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "flags": [], "fullname": "torch.nn.quantized.functional.upsample_bilinear", "name": "upsample_bilinear", "type": null}}, "upsample_nearest": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 1, 1], "arg_names": ["input", "size", "scale_factor"], "flags": [], "fullname": "torch.nn.quantized.functional.upsample_nearest", "name": "upsample_nearest", "type": null}}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/nn/quantized/functional.py"}