diff --git a/fastdeploy/runtime/backends/paddle/option.h b/fastdeploy/runtime/backends/paddle/option.h index fd2975cc9..652b72401 100755 --- a/fastdeploy/runtime/backends/paddle/option.h +++ b/fastdeploy/runtime/backends/paddle/option.h @@ -67,10 +67,12 @@ struct PaddleBackendOption { /// initialize memory size(MB) for GPU int gpu_mem_init_size = 100; + /// Disable type of operators run on TensorRT void DisableTrtOps(const std::vector& ops) { trt_disabled_ops_.insert(trt_disabled_ops_.end(), ops.begin(), ops.end()); } + /// Delete pass by name void DeletePass(const std::string& pass_name) { delete_pass_names.push_back(pass_name); }