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

Fix the instance norm test input size bug in layout infer test (#661) 

Correct gamma and beta size in InstanceNorm.nhwc case

Type: Bug Fix
Issue: 37103

Signed-off-by: Feiyue Chen <Feiyue.Chen@verisilicon.com>
This commit is contained in:
Chen Feiyue 2023-11-22 09:20:27 +08:00 committed by GitHub
parent 74e2740daa
commit 517397949d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 19 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
src/tim/transform/layout_inference_test.cc
View File

@ -229,7 +229,7 @@ TEST(InstanceNorm, nhwc) {
auto src_graph = ctx->CreateGraph();
tim::vx::ShapeType io_shape({2, 2, 2, 2}); //nhwc
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,
@ -245,8 +245,8 @@ TEST(InstanceNorm, nhwc) {
std::vector<float> in_data = {
0.0f, 1.0f, 0.0f, 2.0f, 0.0f, 2.0f, 0.0f, 4.0f, 1.0f, -1.0f, -1.0f, 2.0f, -1.0f, -2.0f, 1.0f, 4.0f
};
std::vector<float> beta = {0};
std::vector<float> gamma = {1.0f};
std::vector<float> beta = {0,0};
std::vector<float> gamma = {1.0f,1.0f};
std::vector<float> golden = {
0.0f, -1.1470304f, 0.0f, -0.22940612f, 0.0f, -0.22940612f, 0.0f, 1.6058424f, 0.99995005f,
-0.7337929f, -0.99995005f, 0.52413774f, -0.99995005f, -1.1531031f, 0.99995005f, 1.3627582f,