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
369889b6
Unverified
Commit
369889b6
authored
May 06, 2022
by
andyjpaddle
Committed by
GitHub
May 06, 2022
Browse files
fix rec v3 mkldnn irpass in cpp infer (#6154)
* fix rec v3 mkldnn irpass * fix rec v3 mkldnn irpass
parent
12aa5e80
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 @
369889b6
...
...
@@ -166,6 +166,10 @@ void CRNNRecognizer::LoadModel(const std::string &model_dir) {
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
);
// true for multiple input
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