colin
/
TIM-VX
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

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>
This commit is contained in:
chxin66 2022-04-13 10:10:30 +08:00 committed by GitHub
parent 1eaf326abf
commit ba6b311409
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 47 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
src/tim/vx/ops/activations_test.cc
View File

@ -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();