1 # flake8: noqa: F401 2 r""" 3 This file is in the process of migration to `torch/ao/quantization`, and 4 is kept here for compatibility while the migration process is ongoing. 5 If you are adding a new entry/functionality, please, add it to the 6 `torch/ao/ns/_numeric_suite.py`, while adding an import statement 7 here. 8 """ 9 10 from torch.ao.ns._numeric_suite import ( 11 _convert_tuple_to_list, 12 _dequantize_tensor_list, 13 _find_match, 14 _get_logger_dict_helper, 15 _is_identical_module_type, 16 compare_model_outputs, 17 compare_model_stub, 18 compare_weights, 19 get_logger_dict, 20 get_matching_activations, 21 Logger, 22 NON_LEAF_MODULE_TO_ADD_OBSERVER_ALLOW_LIST, 23 OutputLogger, 24 prepare_model_outputs, 25 prepare_model_with_stubs, 26 Shadow, 27 ShadowLogger, 28 ) 29