CMakeLists.txt 2.1 KB
Newer Older
Y
Yan Chunwei 已提交
1
# Add TRT tests
2
nv_library(tensorrt_converter
Z
zhupengyang 已提交
3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35
           SRCS matmul_op.cc
                conv2d_op.cc
                fc_op.cc
                pool2d_op.cc
                elementwise_op.cc
                batch_norm_op.cc
                activation_op.cc
                unary_op.cc
                softmax_op.cc
                concat_op.cc
                dropout_op.cc
                group_norm_op.cc
                pad_op.cc
                split_op.cc
                prelu_op.cc
                leaky_relu_op.cc
                gelu_op.cc
                layer_norm_op.cc
                multihead_matmul_op.cc
                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
                emb_eltwise_layernorm.cc
                skip_layernorm.cc
                scale_op.cc
                slice_op.cc
                hard_sigmoid_op.cc
                hard_swish_op.cc
                clip_op.cc
36
                gather_op.cc
37
                anchor_generator_op.cc
Z
zlsh80826 已提交
38
                yolo_box_op.cc
39
                yolo_box_head_op.cc
40
                arg_max_op.cc
41
                roi_align_op.cc
42
                affine_channel_op.cc
Z
zlsh80826 已提交
43
                multiclass_nms_op.cc
44
                multiclass_nms3_op.cc
45
                nearest_interp_op.cc
W
Wangzheee 已提交
46
                reshape_op.cc
47 48
                reduce_op.cc
                gather_nd_op.cc
W
wenbin 已提交
49
                tile_op.cc
W
wenbin 已提交
50
                conv3d_op.cc
W
wangxinxin08 已提交
51
                mish_op.cc
52
                nearest_interp_v2_op.cc
F
feng_shuai 已提交
53
                pool3d_op.cc
W
wangxinxin08 已提交
54
                deformable_conv_op.cc
55
                preln_emb_eltwise_layernorm.cc
F
feng_shuai 已提交
56
		strided_slice_op.cc
57
                preln_skip_layernorm.cc
F
feng_shuai 已提交
58
		roll_op.cc
59 60 61
                transformer_input_convert_op.cc
                remove_padding_op.cc
                recover_padding_op.cc
62
           DEPS tensorrt_engine tensorrt_plugin operator scope framework_proto op_registry)
63 64

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