2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
4 * Licensed under the Apache License, Version 2.0 (the "License");
5 * you may not use this file except in compliance with the License.
6 * You may obtain a copy of the License at
8 * http://www.apache.org/licenses/LICENSE-2.0
10 * Unless required by applicable law or agreed to in writing, software
11 * distributed under the License is distributed on an "AS IS" BASIS,
12 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13 * See the License for the specific language governing permissions and
14 * limitations under the License.
16 #include "kernels/InstanceNorm.h"
17 #include "kernels/TestUtils.h"
18 #include "luci_interpreter/TestMemoryManager.h"
20 namespace luci_interpreter
27 using namespace testing;
29 class InstanceNormTest : public ::testing::Test
32 void SetUp() override { _memory_manager = std::make_unique<TestMemoryManager>(); }
34 std::unique_ptr<IMemoryManager> _memory_manager;
37 TEST_F(InstanceNormTest, Simple)
40 makeInputTensor<DataType::FLOAT32>({1, 2, 2, 1}, {1, 1, 1, 1}, _memory_manager.get());
41 Tensor gamma_tensor = makeInputTensor<DataType::FLOAT32>({1}, {1}, _memory_manager.get());
42 Tensor beta_tensor = makeInputTensor<DataType::FLOAT32>({1}, {2}, _memory_manager.get());
43 Tensor output_tensor = makeOutputTensor(DataType::FLOAT32);
45 InstanceNormParams params{};
46 params.epsilon = 0.1f;
47 params.activation = Activation::NONE;
49 InstanceNorm kernel(&input_tensor, &gamma_tensor, &beta_tensor, &output_tensor, params);
51 _memory_manager->allocate_memory(output_tensor);
54 EXPECT_THAT(extractTensorData<float>(output_tensor), FloatArrayNear({2, 2, 2, 2}));
55 EXPECT_THAT(extractTensorShape(output_tensor), ::testing::ElementsAreArray({1, 2, 2, 1}));
58 TEST_F(InstanceNormTest, Single_gamma_beta)
61 makeInputTensor<DataType::FLOAT32>({1, 2, 1, 2}, {1, 1, 1, 1}, _memory_manager.get());
62 Tensor gamma_tensor = makeInputTensor<DataType::FLOAT32>({1}, {1}, _memory_manager.get());
63 Tensor beta_tensor = makeInputTensor<DataType::FLOAT32>({1}, {2}, _memory_manager.get());
64 Tensor output_tensor = makeOutputTensor(DataType::FLOAT32);
66 InstanceNormParams params{};
67 params.epsilon = 0.1f;
68 params.activation = Activation::NONE;
70 InstanceNorm kernel(&input_tensor, &gamma_tensor, &beta_tensor, &output_tensor, params);
72 _memory_manager->allocate_memory(output_tensor);
75 EXPECT_THAT(extractTensorData<float>(output_tensor), FloatArrayNear({2, 2, 2, 2}));
76 EXPECT_THAT(extractTensorShape(output_tensor), ::testing::ElementsAreArray({1, 2, 1, 2}));
79 TEST_F(InstanceNormTest, Wrong_gamma_beta_dim_NEG)
82 makeInputTensor<DataType::FLOAT32>({1, 2, 1, 2}, {1, 1, 1, 1}, _memory_manager.get());
83 Tensor gamma_tensor = makeInputTensor<DataType::FLOAT32>({3}, {1, 1, 1}, _memory_manager.get());
84 Tensor beta_tensor = makeInputTensor<DataType::FLOAT32>({3}, {2, 2, 2}, _memory_manager.get());
85 Tensor output_tensor = makeOutputTensor(DataType::FLOAT32);
87 InstanceNormParams params{};
88 params.epsilon = 0.1f;
89 params.activation = Activation::NONE;
91 InstanceNorm kernel(&input_tensor, &gamma_tensor, &beta_tensor, &output_tensor, params);
92 EXPECT_ANY_THROW(kernel.configure());
96 } // namespace kernels
97 } // namespace luci_interpreter