1# Copyright (c) Qualcomm Innovation Center, Inc. 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 7from . import ( 8 node_visitor, 9 op_add, 10 op_avg_pool2d, 11 op_batch_norm, 12 op_bmm, 13 op_cat, 14 op_ceil, 15 op_clamp, 16 op_conv2d, 17 op_depth_to_space, 18 op_dequantize, 19 op_div, 20 op_embedding, 21 op_expand, 22 op_gelu, 23 op_group_norm, 24 op_hardsigmoid, 25 op_hardswish, 26 op_hardtanh, 27 op_index, 28 op_index_put, 29 op_layer_norm, 30 op_linear, 31 op_log_softmax, 32 op_matmul, 33 op_max_pool2d, 34 op_mean_dim, 35 op_mul, 36 op_pad, 37 op_pow, 38 op_prelu, 39 op_quantize, 40 op_relu, 41 op_reshape, 42 op_rms_norm, 43 op_rsqrt, 44 op_select_copy, 45 op_sigmoid, 46 op_skip_ops, 47 op_slice_copy, 48 op_softmax, 49 op_space_to_depth, 50 op_split_with_sizes, 51 op_sqrt, 52 op_squeeze, 53 op_sub, 54 op_sum_int_list, 55 op_tanh, 56 op_to, 57 op_topk, 58 op_transpose, 59 op_unsqueeze, 60 op_upsample_bilinear2d, 61 op_upsample_nearest2d, 62) 63 64__all__ = [ 65 node_visitor, 66 op_add, 67 op_avg_pool2d, 68 op_batch_norm, 69 op_bmm, 70 op_cat, 71 op_ceil, 72 op_clamp, 73 op_conv2d, 74 op_depth_to_space, 75 op_dequantize, 76 op_div, 77 op_embedding, 78 op_expand, 79 op_gelu, 80 op_group_norm, 81 op_hardswish, 82 op_hardtanh, 83 op_hardsigmoid, 84 op_index, 85 op_index_put, 86 op_layer_norm, 87 op_linear, 88 op_log_softmax, 89 op_matmul, 90 op_max_pool2d, 91 op_mean_dim, 92 op_mul, 93 op_pad, 94 op_pow, 95 op_prelu, 96 op_quantize, 97 op_relu, 98 op_reshape, 99 op_rms_norm, 100 op_rsqrt, 101 op_select_copy, 102 op_sigmoid, 103 op_skip_ops, 104 op_slice_copy, 105 op_softmax, 106 op_space_to_depth, 107 op_split_with_sizes, 108 op_squeeze, 109 op_sqrt, 110 op_sub, 111 op_sum_int_list, 112 op_tanh, 113 op_topk, 114 op_to, 115 op_transpose, 116 op_unsqueeze, 117 op_upsample_bilinear2d, 118 op_upsample_nearest2d, 119] 120