xref: /aosp_15_r20/external/executorch/backends/xnnpack/__init__.py (revision 523fa7a60841cd1ecfb9cc4201f1ca8b03ed023a)
1# Copyright (c) Meta Platforms, Inc. and affiliates.
2# All rights reserved.
3#
4# This source code is licensed under the BSD-style license found in the
5# LICENSE file in the root directory of this source tree.
6
7# Exposed Partitioners in XNNPACK Package
8from .partition.xnnpack_partitioner import (
9    XnnpackDynamicallyQuantizedPartitioner,
10    XnnpackPartitioner,
11)
12
13# Exposed Configs in XNNPACK Package
14from .utils.configs import (
15    get_xnnpack_capture_config,
16    get_xnnpack_edge_compile_config,
17    get_xnnpack_executorch_backend_config,
18)
19
20# Easy util functions
21from .utils.utils import capture_graph_for_xnnpack
22
23# XNNPACK Backend
24from .xnnpack_preprocess import XnnpackBackend
25
26
27__all__ = [
28    "XnnpackDynamicallyQuantizedPartitioner",
29    "XnnpackPartitioner",
30    "XnnpackBackend",
31    "capture_graph_for_xnnpack",
32    "get_xnnpack_capture_config",
33    "get_xnnpack_edge_compile_config",
34    "get_xnnpack_executorch_backend_config",
35]
36