EdgeBoard支持的算子
更新时间:2024-05-16
EdgeBoard支持FPGA特定算子和PaddleLite的ARM算子,主要支持的算子如下:
OP name | FZ3 | FZ5 | FZ9 |
---|---|---|---|
affine_channel | N | N | N |
affine_grid | N | N | N |
arg_max | Y | Y | Y |
assign_value | A | A | A |
batch_norm | Y | Y | Y |
bilinear_interp | Y | Y | Y |
box_coder | Y | Y | Y |
calib | Y | Y | Y |
cast | A | A | A |
concat | Y | Y | Y |
conv2d | Y | Y | Y |
conv2d_transpose | Y | Y | Y |
density_prior_box | N | N | N |
depthwise_conv2d | Y | Y | Y |
depthwise_conv2d_transpose | N | N | N |
dropout | Y | Y | Y |
elementwise_add | Y | Y | Y |
elementwise_div | N | N | N |
elementwise_max | N | N | N |
elementwise_mod | N | N | N |
elementwise_mul | Y | Y | Y |
elementwise_pow | N | N | N |
elementwise_sub | N | N | N |
elu | N | N | N |
expand | N | N | N |
expand_as | N | N | N |
fc | Y | Y | Y |
feed | Y | Y | Y |
fetch | Y | Y | Y |
fill_constant | Y | Y | Y |
fill_constant_batch_size_like | N | N | N |
flatten | Y | Y | Y |
flatten2 | Y | Y | Y |
fusion_elementwise_add_activation | Y | Y | Y |
fusion_elementwise_div_activation | N | N | N |
fusion_elementwise_max_activation | N | N | N |
fusion_elementwise_mul_activation | N | N | N |
fusion_elementwise_sub_activation | N | N | N |
grid_sampler | N | N | N |
instance_norm | N | N | N |
io_copy | Y | Y | Y |
io_copy_once | Y | Y | Y |
layout | Y | Y | Y |
leaky_relu | Y | Y | Y |
matmul | N | N | N |
mul | Y | Y | Y |
multiclass_nms | Y | Y | Y |
multiclass_nms2 | Y | Y | Y |
nearest_interp | Y | Y | Y |
pad2d | N | N | N |
pool2d | Y | Y | Y |
prelu | N | N | N |
prior_box | Y | Y | Y |
range | N | N | N |
reduce_mean | Y | Y | Y |
relu | Y | Y | Y |
relu6 | Y | Y | Y |
reshape | Y | Y | Y |
reshape2 | Y | Y | Y |
scale | Y | Y | Y |
search_fc | N | N | N |
sequence_topk_avg_pooling | N | N | N |
shuffle_channel | N | N | N |
sigmoid | Y | Y | Y |
slice | Y | Y | Y |
softmax | Y | Y | Y |
split | Y | Y | Y |
squeeze | Y | Y | Y |
squeeze2 | Y | Y | Y |
stack | N | N | N |
subgraph | N | N | N |
tanh | N | N | N |
thresholded_relu | N | N | N |
transpose | Y | Y | Y |
transpose2 | Y | Y | Y |
unsqueeze | Y | Y | Y |
unsqueeze2 | Y | Y | Y |
yolo_box | Y | Y | Y |
Y:代表支持
A:代表arm侧支持
N:代表不支持
更多支持的算子,请参考PaddleLite支持算子列表:https://paddlelite.paddlepaddle.org.cn/introduction/support_operation_list.html