xref: /aosp_15_r20/external/pytorch/torch/quantization/fx/utils.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1# flake8: noqa: F401
2r"""
3This file is in the process of migration to `torch/ao/quantization`, and
4is kept here for compatibility while the migration process is ongoing.
5If you are adding a new entry/functionality, please, add it to the
6appropriate files under `torch/ao/quantization/fx/`, while adding an import statement
7here.
8"""
9from torch.ao.quantization.fx.utils import (
10    all_node_args_have_no_tensors,
11    assert_and_get_unique_device,
12    create_getattr_from_value,
13    get_custom_module_class_keys,
14    get_linear_prepack_op_for_dtype,
15    get_new_attr_name_with_prefix,
16    get_non_observable_arg_indexes_and_types,
17    get_qconv_prepack_op,
18    graph_module_from_producer_nodes,
19    maybe_get_next_module,
20)
21