CMakeLists.txt 1.5 KB
Newer Older
Y
Yan Chunwei 已提交
1
# Add TRT tests
2
nv_library(tensorrt_converter
3
           SRCS matmul_op.cc conv2d_op.cc fc_op.cc pool2d_op.cc elementwise_op.cc
4
                batch_norm_op.cc activation_op.cc softmax_op.cc concat_op.cc dropout_op.cc group_norm_op.cc
P
Pei Yang 已提交
5
                pad_op.cc split_op.cc prelu_op.cc leaky_relu_op.cc gelu_op.cc layer_norm_op.cc multihead_matmul_op.cc
6
                shuffle_channel_op.cc swish_op.cc instance_norm_op.cc stack_op.cc transpose_op.cc flatten_op.cc flatten_contiguous_range_op.cc
P
Pei Yang 已提交
7
                emb_eltwise_layernorm.cc skip_layernorm.cc scale_op.cc slice_op.cc hard_sigmoid_op.cc hard_swish_op.cc clip_op.cc
8
                gather_op.cc
9
                anchor_generator_op.cc
Z
zlsh80826 已提交
10
                yolo_box_op.cc
11
                roi_align_op.cc
12
                affine_channel_op.cc
Z
zlsh80826 已提交
13
                multiclass_nms_op.cc
14
                multiclass_nms3_op.cc
15
                nearest_interp_op.cc
W
Wangzheee 已提交
16
                reshape_op.cc
17 18
                reduce_op.cc
                gather_nd_op.cc
W
wenbin 已提交
19
                tile_op.cc
W
wenbin 已提交
20
                conv3d_op.cc
W
wangxinxin08 已提交
21
                mish_op.cc
22
                nearest_interp_v2_op.cc
F
feng_shuai 已提交
23
                pool3d_op.cc
W
wangxinxin08 已提交
24
                deformable_conv_op.cc
25
                preln_emb_eltwise_layernorm.cc
F
feng_shuai 已提交
26
		strided_slice_op.cc
27
                preln_skip_layernorm.cc
F
feng_shuai 已提交
28
		roll_op.cc
29
           DEPS tensorrt_engine tensorrt_plugin operator scope framework_proto op_registry)
30 31

nv_test(test_op_converter SRCS test_op_converter.cc DEPS
32
  paddle_framework ${GLOB_OPERATOR_DEPS} tensorrt_engine tensorrt_converter)