_paddle_use_ops.h 3.6 KB
Newer Older
Y
Yan Chunwei 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21
// Copyright (c) 2019 PaddlePaddle Authors. All Rights Reserved.
//
// Licensed under the Apache License, Version 2.0 (the "License");
// you may not use this file except in compliance with the License.
// You may obtain a copy of the License at
//
//     http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.

#pragma once

// ATTENTION This can only include in a .cc file.

#include "paddle_lite_factory_helper.h"  // NOLINT

USE_LITE_OP(mul);
W
Wilber 已提交
22
USE_LITE_OP(matmul);
Y
Yan Chunwei 已提交
23
USE_LITE_OP(fc);
T
TianXiaogang 已提交
24
USE_LITE_OP(assign);
Y
Yan Chunwei 已提交
25
USE_LITE_OP(relu);
Y
Yan Chunwei 已提交
26
USE_LITE_OP(relu6);
Y
Yan Chunwei 已提交
27 28 29 30 31 32 33 34 35 36 37 38
USE_LITE_OP(scale);
USE_LITE_OP(feed);
USE_LITE_OP(lrn);
USE_LITE_OP(decode_bboxes);
USE_LITE_OP(box_coder);
USE_LITE_OP(fetch);
USE_LITE_OP(io_copy);
USE_LITE_OP(io_copy_once);
USE_LITE_OP(elementwise_add)
USE_LITE_OP(elementwise_sub)
USE_LITE_OP(elementwise_mul)
USE_LITE_OP(elementwise_max)
39
USE_LITE_OP(elementwise_div)
Y
Yan Chunwei 已提交
40 41 42
USE_LITE_OP(fusion_elementwise_add_activation)
USE_LITE_OP(fusion_elementwise_mul_activation)
USE_LITE_OP(fusion_elementwise_max_activation)
43
USE_LITE_OP(fusion_elementwise_div_activation)
Y
Yan Chunwei 已提交
44 45 46 47 48 49 50
USE_LITE_OP(square)
USE_LITE_OP(softmax)
USE_LITE_OP(dropout)
USE_LITE_OP(concat)
USE_LITE_OP(conv2d)
USE_LITE_OP(depthwise_conv2d)
USE_LITE_OP(pool2d)
51
USE_LITE_OP(max_pool2d_with_index)
Y
Yan Chunwei 已提交
52 53 54 55
USE_LITE_OP(batch_norm)
USE_LITE_OP(fusion_elementwise_sub_activation)
USE_LITE_OP(transpose)
USE_LITE_OP(transpose2)
56
USE_LITE_OP(arg_max)
Y
Yan Chunwei 已提交
57 58 59 60 61 62 63 64
USE_LITE_OP(axpy)
USE_LITE_OP(leaky_relu)
USE_LITE_OP(relu_clipped)
USE_LITE_OP(prelu)
USE_LITE_OP(sigmoid)
USE_LITE_OP(tanh)
USE_LITE_OP(swish)
USE_LITE_OP(log)
Y
Yan Chunwei 已提交
65
USE_LITE_OP(exp)
Y
Yan Chunwei 已提交
66
USE_LITE_OP(conv2d_transpose)
67
USE_LITE_OP(depthwise_conv2d_transpose)
Y
Yan Chunwei 已提交
68 69 70 71 72 73 74
USE_LITE_OP(negative)
USE_LITE_OP(pad2d)
USE_LITE_OP(power)
USE_LITE_OP(shuffle_channel)
USE_LITE_OP(yolo_box)
USE_LITE_OP(bilinear_interp)
USE_LITE_OP(nearest_interp)
75 76
USE_LITE_OP(reduce_mean)
USE_LITE_OP(stack)
Y
Yan Chunwei 已提交
77

78
USE_LITE_OP(assign);
Y
Yan Chunwei 已提交
79 80 81 82 83
USE_LITE_OP(crop)
USE_LITE_OP(prior_box)
USE_LITE_OP(density_prior_box)
USE_LITE_OP(reshape)
USE_LITE_OP(reshape2)
T
TianXiaogang 已提交
84 85
USE_LITE_OP(flatten)
USE_LITE_OP(flatten2)
Y
Yan Chunwei 已提交
86 87 88
USE_LITE_OP(split)
USE_LITE_OP(fake_quantize_moving_average_abs_max);
USE_LITE_OP(fake_dequantize_max_abs);
T
TianXiaogang 已提交
89
USE_LITE_OP(fake_quantize_range_abs_max);
Y
Yan Chunwei 已提交
90 91 92 93 94 95 96 97 98 99 100 101 102 103 104 105 106 107 108 109 110 111 112
USE_LITE_OP(calib);
USE_LITE_OP(calib_once);
USE_LITE_OP(norm);
USE_LITE_OP(layout);
USE_LITE_OP(layout_once);
USE_LITE_OP(im2sequence);
USE_LITE_OP(sequence_softmax);
USE_LITE_OP(logical_xor);
USE_LITE_OP(logical_and);
USE_LITE_OP(less_than);
USE_LITE_OP(top_k);
USE_LITE_OP(increment);
USE_LITE_OP(write_to_array);
USE_LITE_OP(read_from_array);
USE_LITE_OP(gru_unit)
USE_LITE_OP(gru)
USE_LITE_OP(beam_search_decode)
USE_LITE_OP(beam_search)
USE_LITE_OP(fill_constant)
USE_LITE_OP(while)
USE_LITE_OP(lod_reset)
USE_LITE_OP(lookup_table)
USE_LITE_OP(multiclass_nms)
113
USE_LITE_OP(subgraph)
Y
Yan Chunwei 已提交
114 115 116 117 118 119 120
USE_LITE_OP(sequence_expand)
USE_LITE_OP(sequence_pool)
USE_LITE_OP(reduce_max)
USE_LITE_OP(is_empty)
USE_LITE_OP(shape)
USE_LITE_OP(slice)
USE_LITE_OP(cast)
121 122 123
USE_LITE_OP(affine_channel)
USE_LITE_OP(anchor_generator)
USE_LITE_OP(generate_proposals)
124 125 126 127 128
USE_LITE_OP(squeeze)     // for x2paddle
USE_LITE_OP(squeeze2)    // for x2paddle
USE_LITE_OP(unsqueeze)   // for x2paddle
USE_LITE_OP(unsqueeze2)  // for x2paddle
USE_LITE_OP(expand)      // for x2paddle
129
USE_LITE_OP(roi_align)
130
USE_LITE_OP(box_clip)
131 132
USE_LITE_OP(assign_value)
USE_LITE_OP(hard_sigmoid)
J
juncaipeng 已提交
133
USE_LITE_OP(rsqrt)