提交 1795bb1d 编写于 作者: A andyjpaddle

fix rec v3 mkldnn irpass

上级 1090525e
...@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册