aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp
diff options
context:
space:
mode:
authorManuel Bottini <manuel.bottini@arm.com>2019-11-13 17:24:43 +0000
committerManuel Bottini <manuel.bottini@arm.com>2019-11-21 12:18:36 +0000
commit581f178b67a74060fc8a5272c5a6c9571d855131 (patch)
tree5112d3d1fd010df0bd61e8a72c573f625b933651 /src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp
parent2963a6ff8a2fe7bad6c3664935a33ace4d499d61 (diff)
downloadComputeLibrary-581f178b67a74060fc8a5272c5a6c9571d855131.tar.gz
COMPMID-2920: NEInstanceNormalization fails on NHWC validations
Improved TensorInfo to accept DataLayout, useful to test the validate functions Removing nighlies tests Moving all vpadds instructions in add.h Change-Id: I96290a6f26272eae865dba48bbc3c6aee4bc0214 Signed-off-by: Manuel Bottini <manuel.bottini@arm.com> Reviewed-on: https://review.mlplatform.org/c/2287 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com> Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Diffstat (limited to 'src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp')
-rw-r--r--src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp b/src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp
index 295f80af95..d7cb7de627 100644
--- a/src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp
+++ b/src/runtime/NEON/functions/NEInstanceNormalizationLayer.cpp
@@ -64,7 +64,7 @@ void NEInstanceNormalizationLayer::configure(ITensor *input, ITensor *output, fl
Status NEInstanceNormalizationLayer::validate(const ITensorInfo *input, const ITensorInfo *output, float gamma, float beta, float epsilon)
{
- return NEInstanceNormalizationLayerKernel::validate(input, &output->clone()->set_data_layout(DataLayout::NCHW), gamma, beta, epsilon);
+ return NEInstanceNormalizationLayerKernel::validate(&input->clone()->set_data_layout(DataLayout::NCHW), &output->clone()->set_data_layout(DataLayout::NCHW), gamma, beta, epsilon);
}
void NEInstanceNormalizationLayer::run()