Update option.h

This commit is contained in:
Jason
2023-02-22 11:28:04 +08:00
committed by GitHub
parent cc9031c17e
commit dd527388bc

View File

@@ -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);
}