From f9cb2dbe45c30f4096dba06a8e506475b9955d78 Mon Sep 17 00:00:00 2001 From: "Jing.Deng" Date: Fri, 9 Jul 2021 10:34:08 +0800 Subject: [PATCH] fix the axis issue about perchannel quantized conv2d Signed-off-by: Jing.Deng --- src/tim/vx/ops/conv2d_test.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/tim/vx/ops/conv2d_test.cc b/src/tim/vx/ops/conv2d_test.cc index 791024f..39b768e 100644 --- a/src/tim/vx/ops/conv2d_test.cc +++ b/src/tim/vx/ops/conv2d_test.cc @@ -1439,7 +1439,7 @@ TEST(Conv2d, shape_3_2_2_1_int8_QuantizedPerChannelTest) { tim::vx::Quantization quantInput(tim::vx::QuantType::ASYMMETRIC, 2, scalesInput, zeroPointsInput); tim::vx::Quantization quantWeight(tim::vx::QuantType::SYMMETRIC_PER_CHANNEL, - 2, scalesWeight, zeroPointsWeight); + 3, scalesWeight, zeroPointsWeight); tim::vx::Quantization quantBias(tim::vx::QuantType::SYMMETRIC_PER_CHANNEL, 0, scalesBias, zeroPointsBias); tim::vx::Quantization quantOutput(tim::vx::QuantType::ASYMMETRIC, 2,