mirror of
https://github.com/PaddlePaddle/FastDeploy.git
synced 2025-12-24 13:28:13 +08:00
Update option.h
This commit is contained in:
@@ -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<std::string>& 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);
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user