diff --git a/inference-engine/src/offline_transformations/include/pruning.hpp b/inference-engine/src/offline_transformations/include/pruning.hpp index c16433685a8..f398ab1713f 100644 --- a/inference-engine/src/offline_transformations/include/pruning.hpp +++ b/inference-engine/src/offline_transformations/include/pruning.hpp @@ -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 & condition = [](const double & value) { return value < 1e-5; }); + const std::function & condition = [](const double & value) { return value == 0; }); }; /**