natas_hw
/
openvino
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Set default InitConstMask treshold to zero (#5515)

This commit is contained in:
Gleb Kazantaev 2021-05-14 18:47:13 +03:00 committed by GitHub
parent addf7dc1e0
commit 500b922457
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
inference-engine/src/offline_transformations/include/pruning.hpp
View File

@ -32,7 +32,7 @@ class ngraph::pass::InitConstMask : public ngraph::pass::MatcherPass {
public:
NGRAPH_RTTI_DECLARATION;
explicit InitConstMask(const ngraph::AxisSet & dims,
const std::function<bool(const double & value)> & condition = [](const double & value) { return value < 1e-5; });
const std::function<bool(const double & value)> & condition = [](const double & value) { return value == 0; });
};
/**