From dd527388bc227f836612352bfcb99aa82b041fe9 Mon Sep 17 00:00:00 2001 From: Jason Date: Wed, 22 Feb 2023 11:28:04 +0800 Subject: [PATCH] Update option.h --- fastdeploy/runtime/backends/paddle/option.h | 2 ++ 1 file changed, 2 insertions(+) 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); }