From 5668856fc94073afcb3f319bd7e7f961b47d8752 Mon Sep 17 00:00:00 2001 From: Zhouheng Zheng Date: Wed, 30 Aug 2023 14:10:23 +0800 Subject: [PATCH] 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 --- src/tim/vx/ops/instancenormalization_test.cc | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/src/tim/vx/ops/instancenormalization_test.cc b/src/tim/vx/ops/instancenormalization_test.cc index 6160885..f5c703b 100644 --- a/src/tim/vx/ops/instancenormalization_test.cc +++ b/src/tim/vx/ops/instancenormalization_test.cc @@ -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 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 gamma = {1.0f}; - std::vector beta = {0}; + std::vector gamma = {1.0f, 1.0f}; + std::vector beta = {.0f, .0f}; std::vector 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