Commit 4ff6a1bc authored by zuoshaobo's avatar zuoshaobo Committed by Alexander Alekhin

Merge pull request #11425 from zuoshaobo:relu_negative_slope

* FIX INF_ENGINE RELU ERROR

* set slope to variable

* tab in indentwq
parent 684cf433
...@@ -329,6 +329,7 @@ struct ReLUFunctor ...@@ -329,6 +329,7 @@ struct ReLUFunctor
{ {
lp.type = "ReLU"; lp.type = "ReLU";
std::shared_ptr<InferenceEngine::ReLULayer> ieLayer(new InferenceEngine::ReLULayer(lp)); std::shared_ptr<InferenceEngine::ReLULayer> ieLayer(new InferenceEngine::ReLULayer(lp));
ieLayer->negative_slope = slope;
return ieLayer; return ieLayer;
} }
#endif // HAVE_INF_ENGINE #endif // HAVE_INF_ENGINE
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment