diff options
author | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-02 15:08:59 +0100 |
---|---|---|
committer | Narumol Prangnawarat <narumol.prangnawarat@arm.com> | 2019-08-02 15:26:58 +0100 |
commit | ae7b832a6f5eda4b28577f57909111135a36dee9 (patch) | |
tree | 3ea84622726dc3800c741d7be3cebb775eb20bb2 /src/armnn/test/InferOutputTests.hpp | |
parent | 87f65eab4abb65273ea11eb8ca876196ef82c6c5 (diff) | |
download | armnn-ae7b832a6f5eda4b28577f57909111135a36dee9.tar.gz |
IVGCVSW-3604 Fix channel shape calculation in TransposeConvolution2dLayer::InferOutputShapes
Signed-off-by: Narumol Prangnawarat <narumol.prangnawarat@arm.com>
Change-Id: I2e3d5922bb89c8f3b84ff5458fda981ff177c3ce
Diffstat (limited to 'src/armnn/test/InferOutputTests.hpp')
-rw-r--r-- | src/armnn/test/InferOutputTests.hpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/armnn/test/InferOutputTests.hpp b/src/armnn/test/InferOutputTests.hpp index 2dd2ff0e73..c428a9db61 100644 --- a/src/armnn/test/InferOutputTests.hpp +++ b/src/armnn/test/InferOutputTests.hpp @@ -406,7 +406,7 @@ void TransposeConvolution2dInferOutputShapeTest() armnn::TensorShape filterShape(4, filterSize.data()); shapes.push_back(filterShape); - const std::vector<unsigned int> expectedOutputSizes = {1, 2, 6, 6}; + const std::vector<unsigned int> expectedOutputSizes = {1, 1, 6, 6}; armnn::TensorShape expectedOutputShape(4, expectedOutputSizes.data()); BOOST_CHECK(expectedOutputShape == transposeConvolution2dLayer->InferOutputShapes(shapes).at(0)); |