2 * Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
3 * Copyright 2019 The TensorFlow Authors. All Rights Reserved.
5 * Licensed under the Apache License, Version 2.0 (the "License");
6 * you may not use this file except in compliance with the License.
7 * You may obtain a copy of the License at
9 * http://www.apache.org/licenses/LICENSE-2.0
11 * Unless required by applicable law or agreed to in writing, software
12 * distributed under the License is distributed on an "AS IS" BASIS,
13 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14 * See the License for the specific language governing permissions and
15 * limitations under the License.
18 #include "kernels/Rsqrt.h"
19 #include "kernels/TestUtils.h"
20 #include "luci_interpreter/TestMemoryManager.h"
22 namespace luci_interpreter
29 using namespace testing;
31 void Check(std::initializer_list<int32_t> input_shape, std::initializer_list<int32_t> output_shape,
32 std::initializer_list<float> input_data, std::initializer_list<float> output_data)
34 std::unique_ptr<IMemoryManager> memory_manager = std::make_unique<TestMemoryManager>();
37 makeInputTensor<DataType::FLOAT32>(input_shape, input_data, memory_manager.get());
38 Tensor output_tensor = makeOutputTensor(DataType::FLOAT32);
40 Rsqrt kernel(&input_tensor, &output_tensor);
42 memory_manager->allocate_memory(output_tensor);
45 EXPECT_THAT(extractTensorData<float>(output_tensor), FloatArrayNear(output_data));
46 EXPECT_THAT(extractTensorShape(output_tensor), ::testing::ElementsAreArray(output_shape));
49 TEST(RsqrtTest, SimpleRsqrt)
52 /*input_shape=*/{1, 2, 4, 1}, /*output_shape=*/{1, 2, 4, 1},
60 0.44721360, 0.5, 0.35355339, 0.70710678, //
61 0.40824829, 0.36514837, 0.33333333, 1.8257419, //
65 TEST(RsqrtTest, Input_Output_Type_NEG)
67 std::unique_ptr<IMemoryManager> memory_manager = std::make_unique<TestMemoryManager>();
69 Tensor input_tensor = makeInputTensor<DataType::FLOAT32>({1}, {1.f}, memory_manager.get());
70 Tensor output_tensor = makeOutputTensor(DataType::S32);
72 Rsqrt kernel(&input_tensor, &output_tensor);
73 EXPECT_ANY_THROW(kernel.configure());
76 TEST(RsqrtTest, Invalid_Input_Type_NEG)
78 std::unique_ptr<IMemoryManager> memory_manager = std::make_unique<TestMemoryManager>();
80 Tensor input_tensor = makeInputTensor<DataType::S64>({1}, {1}, memory_manager.get());
81 Tensor output_tensor = makeOutputTensor(DataType::S64);
83 Rsqrt kernel(&input_tensor, &output_tensor);
85 memory_manager->allocate_memory(output_tensor);
86 EXPECT_ANY_THROW(kernel.execute());
90 } // namespace kernels
91 } // namespace luci_interpreter