1 // Copyright (C) 2018-2019 Intel Corporation
2 // SPDX-License-Identifier: Apache-2.0
5 #include <gtest/gtest.h>
7 #include <ie_builders.hpp>
8 #include <builders/ie_relu6_layer.hpp>
10 #include "builder_test.hpp"
12 using namespace testing;
13 using namespace InferenceEngine;
15 class ReLU6LayerBuilderTest : public BuilderTestCommon {};
17 TEST_F(ReLU6LayerBuilderTest, getExistsLayerFromNetworkBuilder) {
18 Builder::Network net("network");
19 Builder::ReLU6Layer relu6Layer("relu6layer");
21 size_t ind = net.addLayer(relu6Layer);
22 Builder::ReLU6Layer layerFromNet(net.getLayer(ind));
23 ASSERT_EQ(relu6Layer.getN(), layerFromNet.getN());
26 TEST_F(ReLU6LayerBuilderTest, cannotCreateLayerWithWrongShapes) {
27 Builder::Network net("network");
28 Builder::Layer::Ptr fakeReLU6LayerPtr = std::make_shared<Builder::Layer>("ReLU6", "ReLU6 layer");
29 fakeReLU6LayerPtr->getInputPorts().push_back(Port({1, 1, 1, 1}));
30 fakeReLU6LayerPtr->getOutputPorts().push_back(Port({1, 1, 1, 2}));
31 Builder::ReLU6Layer reLU6Layer(fakeReLU6LayerPtr);
33 ASSERT_THROW(net.addLayer(reLU6Layer), InferenceEngine::details::InferenceEngineException);