支持OP列表
Ops
- argmax
- axpy
- batch_norm
- beam_search
- beam_search_decode
- bilinear_interp
- box_coder
- calib
- calib_once
- cast
- concat
- conv2d
- conv2d_transpose
- crop
- decode_bboxes
- density_prior_box
- depthwise_conv2d
- dropout
- elementwise_add
- elementwise_max
- elementwise_mul
- elementwise_sub
- exp
- fake_dequantize_max_abs
- fake_quantize_moving_average_abs_max
- fc
- feed
- fetch
- fill_constant
- fusion_elementwise_add_activation
- fusion_elementwise_max_activation
- fusion_elementwise_mul_activation
- fusion_elementwise_sub_activation
- graph_op
- gru
- gru_unit
- im2sequence
- increment
- io_copy
- io_copy_once
- is_empty
- layout
- layout_once
- leaky_relu
- less_than
- lod_reset
- log
- logical_and
- logical_xor
- lookup_table
- lrn
- mul
- multiclass_nms
- nearest_interp
- negative
- norm
- pad2d
- pool2d
- power
- prelu
- prior_box
- read_from_array
- reduce_max
- relu
- relu_clipped
- reshape
- reshape2
- scale
- sequence_expand
- sequence_pool
- sequence_softmax
- shape
- shuffle_channel
- sigmoid
- slice
- softmax
- split
- square
- swish
- tanh
- top_k
- transpose
- transpose2
- while
- write_to_array
- yolo_box
Kernels
ARM kernels
- argmax
- axpy
- batch_norm
- beam_search
- beam_search_decode
- bilinear_interp
- box_coder
- calib
- calib_once
- cast
- concat
- conv2d
- conv2d_transpose
- crop
- decode_bboxes
- density_prior_box
- depthwise_conv2d
- dropout
- elementwise_add
- elementwise_max
- elementwise_mul
- exp
- fc
- fill_constant
- fusion_elementwise_add_activation
- fusion_elementwise_max_activation
- fusion_elementwise_mul_activation
- gru
- gru_unit
- im2sequence
- increment
- is_empty
- leaky_relu
- less_than
- lod_reset
- log
- logical_and
- logical_xor
- lookup_table
- lrn
- mul
- multiclass_nms
- nearest_interp
- negative
- norm
- pad2d
- pool2d
- power
- prelu
- prior_box
- read_from_array
- reduce_max
- relu
- relu_clipped
- scale
- sequence_expand
- sequence_pool
- sequence_softmax
- shape
- shuffle_channel
- sigmoid
- slice
- softmax
- split
- swish
- tanh
- top_k
- transpose
- transpose2
- while
- write_to_array
- yolo_box
X86 kernels
- batch_norm
- concat
- conv2d
- depthwise_conv2d
- dropout
- elementwise_add
- elementwise_sub
- fc
- fill_constant
- mul
- pool2d
- relu
- scale
- softmax
- square
OpenCL kernels
- conv2d
- depthwise_conv2d
- elementwise_add
- fc
- fusion_elementwise_add_activation
- io_copy
- io_copy_once
- mul
- pool2d
- relu