aboutsummaryrefslogtreecommitdiff
path: root/src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp
diff options
context:
space:
mode:
authorPablo Tello <pablo.tello@arm.com>2021-04-26 15:39:05 +0100
committerPablo Marquez Tello <pablo.tello@arm.com>2021-04-27 11:29:07 +0000
commit5c3eeec645883dc8f57a1e10995b4e8298343ecb (patch)
tree8c936233e98c2d0032b60cdd1c399bf4643527df /src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp
parent0ff73c70003bad895e1d0bd5c53aa117eb798900 (diff)
downloadComputeLibrary-5c3eeec645883dc8f57a1e10995b4e8298343ecb.tar.gz
Fixed CTS failures CLInstanceNorm
* Resolves COMPMID-4400 Change-Id: I54c33a017c735194fbf4437d1c7df465208bc0ca Signed-off-by: Pablo Tello <pablo.tello@arm.com> Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5505 Comments-Addressed: Arm Jenkins <bsgcomp@arm.com> Tested-by: Arm Jenkins <bsgcomp@arm.com> Reviewed-by: Sheri Zhang <sheri.zhang@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp')
-rw-r--r--src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp2
1 files changed, 1 insertions, 1 deletions
diff --git a/src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp b/src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp
index f2406d68f4..4a0bda8255 100644
--- a/src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp
+++ b/src/runtime/CL/functions/CLInstanceNormalizationLayer.cpp
@@ -52,7 +52,7 @@ void CLInstanceNormalizationLayer::configure(ICLTensor *input, ICLTensor *output
void CLInstanceNormalizationLayer::configure(const CLCompileContext &compile_context, ICLTensor *input, ICLTensor *output, float gamma, float beta, float epsilon, bool use_mixed_precision)
{
auto w = std::make_unique<CLComputeMeanVariance>();
- w->configure(compile_context, input, &_mean_var_tensor);
+ w->configure(compile_context, input, &_mean_var_tensor, use_mixed_precision);
_mean_var_kernel = std::move(w);
auto k = std::make_unique<CLInstanceNormalizationLayerKernel>();
k->configure(compile_context, input, &_mean_var_tensor, output, InstanceNormalizationLayerKernelInfo(gamma, beta, epsilon, use_mixed_precision));