From ba6b3114094eeff873ae4da6ff21c8a51f557808 Mon Sep 17 00:00:00 2001
From: chxin66 <57057788+chxin66@users.noreply.github.com>
Date: Wed, 13 Apr 2022 10:10:30 +0800
Subject: [PATCH] Added hardsigmoid test case with alpha and beta (#356)

https://github.com/VeriSilicon/TIM-VX/issues/306

Signed-off-by: Chen Xin <jack.chen@verisilicon.com>
---
 src/tim/vx/ops/activations_test.cc | 47 ++++++++++++++++++++++++++++++
 1 file changed, 47 insertions(+)

diff --git a/src/tim/vx/ops/activations_test.cc b/src/tim/vx/ops/activations_test.cc
index 889e134..0470204 100644
--- a/src/tim/vx/ops/activations_test.cc
+++ b/src/tim/vx/ops/activations_test.cc
@@ -226,6 +226,53 @@ TEST(HardSigmoid, shape_5_1_uint8_Quantized) {
   EXPECT_TRUE(ArraysMatch(golden_data, output, (uint8_t)1));
 }
 
+TEST(HardSigmoid, a_b_) {
+  auto ctx = tim::vx::Context::Create();
+  auto graph = ctx->CreateGraph();
+
+  tim::vx::ShapeType in_shape({20, 1});
+  tim::vx::ShapeType out_shape({20, 1});
+
+  std::vector<float> scalesInput = {0.00228914};  //scale
+  std::vector<int32_t> zeroPointsInput = {128};   //zero point
+
+  std::vector<float> scalesOutput = {0.005};
+  std::vector<int32_t> zeroPointsOutput = {128};
+
+  tim::vx::Quantization quantInput(tim::vx::QuantType::ASYMMETRIC, 1,
+                                   scalesInput, zeroPointsInput);
+  tim::vx::Quantization quantOutput(tim::vx::QuantType::ASYMMETRIC, 1,
+                                    scalesOutput, zeroPointsOutput);
+
+  tim::vx::TensorSpec input_spec(tim::vx::DataType::UINT8, in_shape,
+                                 tim::vx::TensorAttribute::INPUT, quantInput);
+
+  tim::vx::TensorSpec output_spec(tim::vx::DataType::UINT8, out_shape,
+                                  tim::vx::TensorAttribute::OUTPUT,
+                                  quantOutput);
+
+  auto input_tensor = graph->CreateTensor(input_spec);
+  auto output_tensor = graph->CreateTensor(output_spec);
+
+  std::vector<uint8_t> in_data = {65,  255, 140, 92,  142, 122, 117,
+                                  167, 132, 117, 44,  99,  109, 96,
+                                  216, 222, 135, 126, 113, 100};
+  std::vector<uint8_t> golden_data = {239, 255, 250, 243, 250, 247, 246,
+                                      253, 249, 246, 236, 244, 245, 244,
+                                      255, 255, 249, 248, 246, 244};
+
+  EXPECT_TRUE(input_tensor->CopyDataToTensor(in_data.data(), in_data.size()));
+  auto op = graph->CreateOperation<tim::vx::ops::HardSigmoid>(0.3, 0.6);
+  (*op).BindInput(input_tensor).BindOutput(output_tensor);
+
+  EXPECT_TRUE(graph->Compile());
+  EXPECT_TRUE(graph->Run());
+  std::vector<uint8_t> output(golden_data.size());
+
+  EXPECT_TRUE(output_tensor->CopyDataFromTensor(output.data()));
+  EXPECT_EQ(golden_data, output);
+}
+
 TEST(Elu, shape_5_1_fp32) {
   auto ctx = tim::vx::Context::Create();
   auto graph = ctx->CreateGraph();