Fork自 PaddlePaddle / Paddle
diff --git a/paddle/fluid/inference/api/analysis_predictor.cc b/paddle/fluid/inference/api/analysis_predictor.cc index 0645af611b..6b05a7fffb 100644 --- a/paddle/fluid/inference/api/analysis_predictor.cc +++ b/paddle/fluid/inference/api/analysis_predictor.cc @@ -1923,7 +1923,7 @@ USE_TRT_CONVERTER(shuffle_channel); USE_TRT_CONVERTER(swish); USE_TRT_CONVERTER(group_norm); USE_TRT_CONVERTER(instance_norm); -USE_TRT_CONVERTER(layer_norm); +//USE_TRT_CONVERTER(layer_norm); USE_TRT_CONVERTER(gelu); USE_TRT_CONVERTER(multihead_matmul); USE_TRT_CONVERTER(fused_embedding_eltwise_layernorm); @@ -1933,13 +1933,13 @@ USE_TRT_CONVERTER(scale); USE_TRT_CONVERTER(stack); USE_TRT_CONVERTER(clip); USE_TRT_CONVERTER(gather); -USE_TRT_CONVERTER(anchor_generator); +//USE_TRT_CONVERTER(anchor_generator); USE_TRT_CONVERTER(yolo_box); USE_TRT_CONVERTER(yolo_box_head); USE_TRT_CONVERTER(arg_max); -USE_TRT_CONVERTER(roi_align); -USE_TRT_CONVERTER(affine_channel); -USE_TRT_CONVERTER(multiclass_nms); +//USE_TRT_CONVERTER(roi_align); +//USE_TRT_CONVERTER(affine_channel); +//USE_TRT_CONVERTER(multiclass_nms); USE_TRT_CONVERTER(multiclass_nms3); USE_TRT_CONVERTER(nearest_interp); USE_TRT_CONVERTER(nearest_interp_v2);