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

Fix the instance norm test input size bug (#645) 

Only NNAPI instance norm spec have scalar gamma and beta, which can not
support by sp, rewrite it into tensor.

Type: Bug Fix

Co-authored-by: zhouheng.zheng <zhouheng.zheng@ouotlook.com>
This commit is contained in:
Zhouheng Zheng 2023-08-30 14:10:23 +08:00 committed by GitHub
parent 3bbe2ef9ec
commit 5668856fc9
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 3 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tim/vx/ops/instancenormalization_test.cc
View File

@ -32,7 +32,7 @@ TEST(InstanceNorm, shape_2_2_2_2_float) {
auto graph = ctx->CreateGraph();
tim::vx::ShapeType io_shape({2, 2, 2, 2}); //nchw
tim::vx::ShapeType param_shape({1});
tim::vx::ShapeType param_shape({2});
tim::vx::TensorSpec input_spec(tim::vx::DataType::FLOAT32,
io_shape, tim::vx::TensorAttribute::INPUT);
tim::vx::TensorSpec param_spec(tim::vx::DataType::FLOAT32,
@ -48,8 +48,8 @@ TEST(InstanceNorm, shape_2_2_2_2_float) {
std::vector<float> in_data = {
0.0f, 0.0f, 0.0f, 0.0f, 1.0f, 2.0f, 2.0f, 4.0f, 1.0f, -1.0f, -1.0f, 1.0f, -1.0f, 2.0f, -2.0f, 4.0f
};
std::vector<float> gamma = {1.0f};
std::vector<float> beta = {0};
std::vector<float> gamma = {1.0f, 1.0f};
std::vector<float> beta = {.0f, .0f};
std::vector<float> golden = {
0.0f, 0.0f, 0.0f, 0.0f, -1.1470304f, -0.22940612f, -0.22940612f, 1.6058424f, 0.99995005f,
-0.99995005f, -0.99995005f, 0.99995005f, -0.7337929f, 0.52413774f, -1.1531031f, 1.3627582f