Skip to content

GitLab

Commit 1795bb1d authored by andyjpaddle's avatar andyjpaddle
Browse files

fix rec v3 mkldnn irpass

parent 1090525e
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
deploy/cpp_infer/src/ocr_rec.cpp
View file @ 1795bb1d
...@@ -166,9 +166,9 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) { ...@@ -166,9 +166,9 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
} }
// 得到 pass_builder 对象 // get pass_builder object
auto pass_builder = config.pass_builder(); auto pass_builder = config.pass_builder();
// 在 IR 优化阶段，去除 fc_fuse_pass // delete "matmul_transpose_reshape_fuse_pass"
pass_builder->DeletePass("matmul_transpose_reshape_fuse_pass"); pass_builder->DeletePass("matmul_transpose_reshape_fuse_pass");
config.SwitchUseFeedFetchOps(false); config.SwitchUseFeedFetchOps(false);
// true for multiple input // true for multiple input
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment