executorch/backends/arm/operators/__init__.py at 9d0be16e4fcb6b4e62e66cfecec2406f3961bd4d · robell/executorch · GitHub

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
# Copyright 2023-2026 Arm Limited and/or its affiliates.
#
# This source code is licensed under the BSD-style license found in the
# LICENSE file in the root directory of this source tree.
"""Import and register Arm TOSA operator visitors.

Importing this package loads all visitor modules so their classes can be
registered via decorators and discovered at runtime.

"""


from . import (  # noqa
    node_visitor,
    op_abs,
    op_add,
    op_amax,
    op_amin,
    op_any,
    op_avg_pool2d,
    op_bitwise_not,
    op_cat,
    op_ceil,
    op_clamp,
    op_cond_if,
    op_cos,
    op_tosa_custom,
    op_eq,
    op_erf,
    op_exp,
    op_floor,
    op_ge,
    op_gt,
    op_log,
    op_logical_not,
    op_max_pool2d,
    op_maximum,
    op_minimum,
    op_mul,
    op_neg,
    op_permute,
    op_pow,
    op_reciprocal,
    op_repeat,
    op_rshift_tensor,
    op_rsqrt,
    op_sigmoid,
    op_sin,
    op_sub,
    op_sum,
    op_tanh,
    op_to_dim_order_copy,
    op_tosa_conv2d,
    op_tosa_conv3d,
    op_tosa_depthwise_conv2d,
    op_tosa_gather,
    op_tosa_matmul,
    op_tosa_pad,
    op_tosa_rescale,
    op_tosa_resize,
    op_tosa_scatter,
    op_tosa_shapes,
    op_tosa_slice,
    op_tosa_table,
    op_tosa_transpose,
    op_tosa_transpose_conv2d,
    op_view,
    op_where,
    op_while,
    ops_binary,
)