__init__.py 10.1 KB
Newer Older
1 2 3 4 5 6 7 8 9 10 11 12 13 14
#   Copyright (c) 2020 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.

15
# TODO: import all neural network related api under this directory,
16
# including layers, linear, conv, rnn etc.
L
Li Fuchen 已提交
17
__all__ = []
18 19

# TODO: define alias in functional directory
20
from . import conv
L
Li Fuchen 已提交
21
__all__ += conv.__all__
22
from . import activation
L
Li Fuchen 已提交
23
__all__ += activation.__all__
24 25 26 27
from . import extension
__all__ += extension.__all__
from . import common
__all__ += common.__all__
28 29
from . import pooling
__all__ += pooling.__all__
30 31
from . import loss
__all__ += loss.__all__
32 33 34
from .activation import elu  #DEFINE_ALIAS
from .activation import erf  #DEFINE_ALIAS
from .activation import gelu  #DEFINE_ALIAS
35
from .activation import hardshrink  #DEFINE_ALIAS
36
from .activation import hardtanh  #DEFINE_ALIAS
37 38
from .activation import hardsigmoid  #DEFINE_ALIAS
from .activation import hardswish  #DEFINE_ALIAS
39
from .activation import leaky_relu  #DEFINE_ALIAS
40
from .activation import log_sigmoid  #DEFINE_ALIAS
41
from .activation import maxout  #DEFINE_ALIAS
42
from .activation import prelu  #DEFINE_ALIAS
43
from .activation import relu  #DEFINE_ALIAS
44 45
from .activation import relu6  #DEFINE_ALIAS
from .activation import selu  #DEFINE_ALIAS
46
from .activation import sigmoid  #DEFINE_ALIAS
47 48 49 50 51 52
from .activation import soft_relu  #DEFINE_ALIAS
from .activation import softmax  #DEFINE_ALIAS
from .activation import softplus  #DEFINE_ALIAS
from .activation import softshrink  #DEFINE_ALIAS
from .activation import softsign  #DEFINE_ALIAS
from .activation import swish  #DEFINE_ALIAS
W
WangXi 已提交
53
from .activation import tanh  #DEFINE_ALIAS
54
from .activation import tanhshrink  #DEFINE_ALIAS
55
from .activation import thresholded_relu  #DEFINE_ALIAS
56
from .activation import log_softmax  #DEFINE_ALIAS
57
from .common import dropout  #DEFINE_ALIAS
58 59
from .common import dropout2d  #DEFINE_ALIAS
from .common import dropout3d  #DEFINE_ALIAS
60
from .common import alpha_dropout  #DEFINE_ALIAS
61
# from .common import embedding        #DEFINE_ALIAS
62
# from .common import fc  #DEFINE_ALIAS
63
from .common import label_smooth
64 65
from .common import one_hot  #DEFINE_ALIAS
from .common import pad  #DEFINE_ALIAS
66
from .common import pad_constant_like  #DEFINE_ALIAS
67
from .common import pad2d  #DEFINE_ALIAS
L
littletomatodonkey 已提交
68
from .common import cosine_similarity  #DEFINE_ALIAS
69 70 71 72
from .common import unfold  #DEFINE_ALIAS
# from .common import bilinear_tensor_product        #DEFINE_ALIAS
from .common import assign  #DEFINE_ALIAS
from .common import interpolate  #DEFINE_ALIAS
73
from .common import upsample  #DEFINE_ALIAS
74
from .common import bilinear  #DEFINE_ALIAS
W
whs 已提交
75
from .conv import conv1d  #DEFINE_ALIAS
76
from .conv import conv_transpose1d  #DEFINE_ALIAS
77
from .common import linear  #DEFINE_ALIAS
78
from .conv import conv2d  #DEFINE_ALIAS
L
LielinJiang 已提交
79
from .conv import conv_transpose2d  #DEFINE_ALIAS
80
from .conv import conv3d  #DEFINE_ALIAS
L
LielinJiang 已提交
81
from .conv import conv_transpose3d  #DEFINE_ALIAS
82 83
from .extension import add_position_encoding  #DEFINE_ALIAS
# from .extension import autoincreased_step_counter        #DEFINE_ALIAS
84 85
from .extension import continuous_value_model  #DEFINE_ALIAS
from .extension import filter_by_instag  #DEFINE_ALIAS
86 87 88
# from .extension import linear_chain_crf        #DEFINE_ALIAS
# from .extension import merge_selected_rows        #DEFINE_ALIAS
from .extension import multiclass_nms  #DEFINE_ALIAS
89 90
from .extension import polygon_box_transform  #DEFINE_ALIAS
from .extension import random_crop  #DEFINE_ALIAS
91
from .extension import row_conv  #DEFINE_ALIAS
92 93
from .extension import rpn_target_assign  #DEFINE_ALIAS
from .extension import similarity_focus  #DEFINE_ALIAS
94 95
from .extension import target_assign  #DEFINE_ALIAS
from .extension import temporal_shift  #DEFINE_ALIAS
96
from .extension import warpctc  #DEFINE_ALIAS
L
Li Fuchen 已提交
97
from .extension import diag_embed  #DEFINE_ALIAS
98 99 100 101 102 103 104 105 106 107 108 109 110 111 112 113 114 115 116 117 118 119 120 121 122 123 124 125 126
# from .lod import sequence_concat        #DEFINE_ALIAS
# from .lod import sequence_conv        #DEFINE_ALIAS
# from .lod import sequence_enumerate        #DEFINE_ALIAS
# from .lod import sequence_expand_as        #DEFINE_ALIAS
# from .lod import sequence_expand        #DEFINE_ALIAS
# from .lod import sequence_first_step        #DEFINE_ALIAS
# from .lod import sequence_last_step        #DEFINE_ALIAS
# from .lod import sequence_mask        #DEFINE_ALIAS
# from .lod import sequence_pad        #DEFINE_ALIAS
# from .lod import sequence_pool        #DEFINE_ALIAS
# from .lod import sequence_reshape        #DEFINE_ALIAS
# from .lod import sequence_reverse        #DEFINE_ALIAS
# from .lod import sequence_scatter        #DEFINE_ALIAS
# from .lod import sequence_slice        #DEFINE_ALIAS
# from .lod import sequence_softmax        #DEFINE_ALIAS
# from .lod import sequence_unpad        #DEFINE_ALIAS
# from .lod import array_length        #DEFINE_ALIAS
# from .lod import array_read        #DEFINE_ALIAS
# from .lod import array_write        #DEFINE_ALIAS
# from .lod import create_array        #DEFINE_ALIAS
from .lod import hash  #DEFINE_ALIAS
# from .lod import im2sequence        #DEFINE_ALIAS
# from .lod import lod_append        #DEFINE_ALIAS
# from .lod import lod_reset        #DEFINE_ALIAS
# from .lod import reorder_lod_tensor_by_rank        #DEFINE_ALIAS
# from .lod import tensor_array_to_tensor        #DEFINE_ALIAS
# from .lod import dynamic_gru        #DEFINE_ALIAS
# from .lod import dynamic_lstm        #DEFINE_ALIAS
# from .lod import dynamic_lstmp        #DEFINE_ALIAS
127
from .loss import binary_cross_entropy  #DEFINE_ALIAS
128
from .loss import binary_cross_entropy_with_logits  #DEFINE_ALIAS
129 130 131 132
from .loss import bpr_loss  #DEFINE_ALIAS
from .loss import center_loss  #DEFINE_ALIAS
from .loss import cross_entropy  #DEFINE_ALIAS
from .loss import dice_loss  #DEFINE_ALIAS
133
from .loss import edit_distance  #DEFINE_ALIAS
134
from .loss import hsigmoid_loss  #DEFINE_ALIAS
135
from .loss import iou_similarity  #DEFINE_ALIAS
136
from .loss import kl_div  #DEFINE_ALIAS
137
from .loss import l1_loss  #DEFINE_ALIAS
138
from .loss import log_loss  #DEFINE_ALIAS
139
from .loss import margin_ranking_loss  #DEFINE_ALIAS
140
from .loss import mse_loss  #DEFINE_ALIAS
141
from .loss import nll_loss  #DEFINE_ALIAS
142 143 144
# from .loss import nce        #DEFINE_ALIAS
from .loss import npair_loss  #DEFINE_ALIAS
from .loss import rank_loss  #DEFINE_ALIAS
145
from .loss import sampled_softmax_with_cross_entropy  #DEFINE_ALIAS
146 147 148
from .loss import sigmoid_cross_entropy_with_logits  #DEFINE_ALIAS
from .loss import sigmoid_focal_loss  #DEFINE_ALIAS
from .loss import smooth_l1  #DEFINE_ALIAS
149
from .loss import smooth_l1_loss  #DEFINE_ALIAS
150 151 152 153
from .loss import softmax_with_cross_entropy  #DEFINE_ALIAS
from .loss import square_error_cost  #DEFINE_ALIAS
from .loss import ssd_loss  #DEFINE_ALIAS
from .loss import teacher_student_sigmoid_loss  #DEFINE_ALIAS
154
from .loss import ctc_loss  #DEFINE_ALIAS
155 156
# from .norm import data_norm        #DEFINE_ALIAS
# from .norm import group_norm        #DEFINE_ALIAS
157 158 159
from .norm import batch_norm  #DEFINE_ALIAS
from .norm import instance_norm  #DEFINE_ALIAS
from .norm import layer_norm  #DEFINE_ALIAS
160
from .norm import local_response_norm  #DEFINE_ALIAS
161
from .norm import normalize  #DEFINE_ALIAS
162 163 164
# from .norm import spectral_norm        #DEFINE_ALIAS
from .pooling import pool2d  #DEFINE_ALIAS
from .pooling import pool3d  #DEFINE_ALIAS
165
from .pooling import avg_pool1d  #DEFINE_ALIAS
166 167
from .pooling import avg_pool2d  #DEFINE_ALIAS
from .pooling import avg_pool3d  #DEFINE_ALIAS
168 169
from .pooling import max_pool1d  #DEFINE_ALIAS
from .pooling import max_pool2d  #DEFINE_ALIAS
170
from .pooling import max_pool3d  #DEFINE_ALIAS
171 172 173 174 175

from .pooling import adaptive_max_pool1d  #DEFINE_ALIAS
from .pooling import adaptive_max_pool2d  #DEFINE_ALIAS
from .pooling import adaptive_max_pool3d  #DEFINE_ALIAS
from .pooling import adaptive_avg_pool1d  #DEFINE_ALIAS
176 177
from .pooling import adaptive_avg_pool2d  #DEFINE_ALIAS
from .pooling import adaptive_avg_pool3d  #DEFINE_ALIAS
178 179 180

from .rnn import rnn  #DEFINE_ALIAS
from .rnn import birnn  #DEFINE_ALIAS
181 182 183 184 185 186 187 188 189 190 191
# from .rnn import gru_unit        #DEFINE_ALIAS
# from .rnn import lstm        #DEFINE_ALIAS
# from .rnn import lstm_unit        #DEFINE_ALIAS
from .vision import affine_channel  #DEFINE_ALIAS
from .vision import affine_grid  #DEFINE_ALIAS
from .vision import anchor_generator  #DEFINE_ALIAS
from .vision import bipartite_match  #DEFINE_ALIAS
from .vision import box_clip  #DEFINE_ALIAS
from .vision import box_coder  #DEFINE_ALIAS
from .vision import box_decoder_and_assign  #DEFINE_ALIAS
from .vision import collect_fpn_proposals  #DEFINE_ALIAS
R
ruri 已提交
192
# from .vision import deformable_conv  #DEFINE_ALIAS
193 194 195 196
from .vision import deformable_roi_pooling  #DEFINE_ALIAS
from .vision import density_prior_box  #DEFINE_ALIAS
from .vision import detection_output  #DEFINE_ALIAS
from .vision import distribute_fpn_proposals  #DEFINE_ALIAS
197
from .vision import fsp_matrix  #DEFINE_ALIAS
198 199 200
from .vision import generate_mask_labels  #DEFINE_ALIAS
from .vision import generate_proposal_labels  #DEFINE_ALIAS
from .vision import generate_proposals  #DEFINE_ALIAS
201
from .vision import grid_sample  #DEFINE_ALIAS
202
from .vision import image_resize_short  #DEFINE_ALIAS
R
ruri 已提交
203
# from .vision import multi_box_head  #DEFINE_ALIAS
204
from .vision import pixel_shuffle  #DEFINE_ALIAS
205 206 207
from .vision import prior_box  #DEFINE_ALIAS
from .vision import prroi_pool  #DEFINE_ALIAS
from .vision import psroi_pool  #DEFINE_ALIAS
208 209
from .vision import retinanet_detection_output  #DEFINE_ALIAS
from .vision import retinanet_target_assign  #DEFINE_ALIAS
210
from .vision import roi_align  #DEFINE_ALIAS
211
from .vision import roi_perspective_transform  #DEFINE_ALIAS
212
from .vision import roi_pool  #DEFINE_ALIAS
213
from .vision import shuffle_channel  #DEFINE_ALIAS
214 215 216
from .vision import space_to_depth  #DEFINE_ALIAS
from .vision import yolo_box  #DEFINE_ALIAS
from .vision import yolov3_loss  #DEFINE_ALIAS
217
from .input import one_hot  #DEFINE_ALIAS
T
tangwei12 已提交
218
from .input import embedding  #DEFINE_ALIAS