| |
| r""" |
| This file is in the process of migration to `torch/ao/quantization`, and |
| is kept here for compatibility while the migration process is ongoing. |
| If you are adding a new entry/functionality, please, add it to the |
| `torch/ao/ns/_numeric_suite_fx.py`, while adding an import statement |
| here. |
| """ |
|
|
| from torch.ao.ns._numeric_suite_fx import ( |
| RNNReturnType, |
| OutputLogger, |
| NSTracer, |
| _extract_weights_one_model, |
| _extract_weights_impl, |
| extract_weights, |
| _add_loggers_one_model, |
| _add_loggers_impl, |
| add_loggers, |
| _extract_logger_info_one_model, |
| extract_logger_info, |
| _add_shadow_loggers_impl, |
| add_shadow_loggers, |
| extract_shadow_logger_info, |
| extend_logger_results_with_comparison, |
| ) |
|
|
