You need to sign in or sign up before continuing.
Unverified Commit 369889b6 authored by andyjpaddle's avatar andyjpaddle Committed by GitHub
Browse files

fix rec v3 mkldnn irpass in cpp infer (#6154)

* fix rec v3 mkldnn irpass

* fix rec v3 mkldnn irpass
parent 12aa5e80
...@@ -166,6 +166,10 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) { ...@@ -166,6 +166,10 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_); config.SetCpuMathLibraryNumThreads(this->cpu_math_library_num_threads_);
} }
// get pass_builder object
auto pass_builder = config.pass_builder();
// delete "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
config.SwitchSpecifyInputNames(true); config.SwitchSpecifyInputNames(true);
......
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