Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
wangsen
paddle_dbnet
Commits
cd21ea87
Commit
cd21ea87
authored
May 06, 2022
by
andyjpaddle
Browse files
fix rec v3 mkldnn irpass
parent
4cdad758
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
4 additions
and
0 deletions
+4
-0
deploy/cpp_infer/src/ocr_rec.cpp
deploy/cpp_infer/src/ocr_rec.cpp
+4
-0
No files found.
deploy/cpp_infer/src/ocr_rec.cpp
View file @
cd21ea87
...
@@ -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_
);
}
}
// 得到 pass_builder 对象
auto
pass_builder
=
config
.
pass_builder
();
// 在 IR 优化阶段,去除 fc_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
);
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment