[SAMPLES] Use defined constant instead of string (#788)
authorIlya Churaev <ilya.churaev@intel.com>
Fri, 5 Jun 2020 08:22:24 +0000 (11:22 +0300)
committerGitHub <noreply@github.com>
Fri, 5 Jun 2020 08:22:24 +0000 (11:22 +0300)
inference-engine/samples/hello_reshape_ssd/reshape_ssd_extension.hpp

index ce27793..edabbfc 100644 (file)
@@ -127,7 +127,7 @@ constexpr ngraph::NodeTypeInfo CustomReluOp::type_info;
 class InPlaceExtension : public InferenceEngine::IExtension {
 public:
     InPlaceExtension() {
-        impls["CustomReLU"] = [](const std::shared_ptr<ngraph::Node>& node) -> InferenceEngine::ILayerImpl::Ptr {
+        impls[CUSTOM_RELU_TYPE] = [](const std::shared_ptr<ngraph::Node>& node) -> InferenceEngine::ILayerImpl::Ptr {
             return std::make_shared<CustomReLUImpl>(node);
         };
     }