icefall/.mypy_cache/3.9/torch/ao/quantization/utils.data.json

{".class": "MypyFile", "_fullname": "torch.ao.quantization.utils", "is_partial_stub_package": false, "is_stub": false, "names": {".class": "SymbolTable", "Any": {".class": "SymbolTableNode", "cross_ref": "typing.Any", "kind": "Gdef"}, "QuantType": {".class": "SymbolTableNode", "cross_ref": "torch.ao.quantization.quant_type.QuantType", "kind": "Gdef"}, "Tuple": {".class": "SymbolTableNode", "cross_ref": "typing.Tuple", "kind": "Gdef"}, "__doc__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.ao.quantization.utils.__doc__", "name": "__doc__", "type": "builtins.str"}}, "__file__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.ao.quantization.utils.__file__", "name": "__file__", "type": "builtins.str"}}, "__name__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.ao.quantization.utils.__name__", "name": "__name__", "type": "builtins.str"}}, "__package__": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "Var", "flags": ["is_ready"], "fullname": "torch.ao.quantization.utils.__package__", "name": "__package__", "type": "builtins.str"}}, "activation_dtype": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.activation_dtype", "name": "activation_dtype", "type": null}}, "activation_is_int8_quantized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.activation_is_int8_quantized", "name": "activation_is_int8_quantized", "type": null}}, "activation_is_statically_quantized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.activation_is_statically_quantized", "name": "activation_is_statically_quantized", "type": null}}, "calculate_qmin_qmax": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0, 0, 0], "arg_names": ["quant_min", "quant_max", "has_customized_qrange", "dtype", "reduce_range"], "flags": [], "fullname": "torch.ao.quantization.utils.calculate_qmin_qmax", "name": "calculate_qmin_qmax", "type": {".class": "CallableType", "arg_kinds": [0, 0, 0, 0, 0], "arg_names": ["quant_min", "quant_max", "has_customized_qrange", "dtype", "reduce_range"], "arg_types": ["builtins.int", "builtins.int", "builtins.bool", "torch._C.dtype", "builtins.bool"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "calculate_qmin_qmax", "ret_type": {".class": "TupleType", "implicit": false, "items": ["builtins.int", "builtins.int"], "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": []}}}, "check_min_max_valid": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["min_val", "max_val"], "flags": [], "fullname": "torch.ao.quantization.utils.check_min_max_valid", "name": "check_min_max_valid", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["min_val", "max_val"], "arg_types": ["torch._tensor.Tensor", "torch._tensor.Tensor"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "check_min_max_valid", "ret_type": "builtins.bool", "type_guard": null, "variables": []}}}, "functools": {".class": "SymbolTableNode", "cross_ref": "functools", "kind": "Gdef"}, "get_combined_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["default_dict", "additional_dict"], "flags": [], "fullname": "torch.ao.quantization.utils.get_combined_dict", "name": "get_combined_dict", "type": null}}, "get_qconfig_dtypes": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.get_qconfig_dtypes", "name": "get_qconfig_dtypes", "type": null}}, "get_qparam_dict": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["observer_or_fake_quant"], "flags": [], "fullname": "torch.ao.quantization.utils.get_qparam_dict", "name": "get_qparam_dict", "type": null}}, "get_quant_type": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.get_quant_type", "name": "get_quant_type", "type": null}}, "get_swapped_custom_module_class": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0, 0], "arg_names": ["custom_module", "custom_module_class_mapping", "qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.get_swapped_custom_module_class", "name": "get_swapped_custom_module_class", "type": null}}, "getattr_from_fqn": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0, 0], "arg_names": ["obj", "fqn"], "flags": [], "fullname": "torch.ao.quantization.utils.getattr_from_fqn", "name": "getattr_from_fqn", "type": {".class": "CallableType", "arg_kinds": [0, 0], "arg_names": ["obj", "fqn"], "arg_types": [{".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "builtins.str"], "bound_args": [], "def_extras": {"first_arg": null}, "fallback": "builtins.function", "implicit": false, "is_ellipsis_args": false, "name": "getattr_from_fqn", "ret_type": {".class": "AnyType", "missing_import_name": null, "source_any": null, "type_of_any": 2}, "type_guard": null, "variables": []}}}, "is_per_channel": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qscheme"], "flags": [], "fullname": "torch.ao.quantization.utils.is_per_channel", "name": "is_per_channel", "type": null}}, "is_per_tensor": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qscheme"], "flags": [], "fullname": "torch.ao.quantization.utils.is_per_tensor", "name": "is_per_tensor", "type": null}}, "quant_type_to_str": {".class": "SymbolTableNode", "cross_ref": "torch.ao.quantization.quant_type.quant_type_to_str", "kind": "Gdef"}, "torch": {".class": "SymbolTableNode", "cross_ref": "torch", "kind": "Gdef"}, "warnings": {".class": "SymbolTableNode", "cross_ref": "warnings", "kind": "Gdef"}, "weight_dtype": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.weight_dtype", "name": "weight_dtype", "type": null}}, "weight_is_quantized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.weight_is_quantized", "name": "weight_is_quantized", "type": null}}, "weight_is_statically_quantized": {".class": "SymbolTableNode", "kind": "Gdef", "node": {".class": "FuncDef", "arg_kinds": [0], "arg_names": ["qconfig"], "flags": [], "fullname": "torch.ao.quantization.utils.weight_is_statically_quantized", "name": "weight_is_statically_quantized", "type": null}}}, "path": "/home/patrick/anaconda3/lib/python3.9/site-packages/torch/ao/quantization/utils.py"}