diff options
author | Dana Zlotnik <dana.zlotnik@arm.com> | 2022-01-03 10:59:41 +0200 |
---|---|---|
committer | Dana Zlotnik <dana.zlotnik@arm.com> | 2022-01-12 12:31:14 +0000 |
commit | d7e2ec51239e2075f931e0a9364e0a68534676f1 (patch) | |
tree | fa2e37fdd125caba983d3b9cdabdb8e03921f6a6 /src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h | |
parent | 5ae8d804d67f57fbfa793800ddcc21a5aff954dd (diff) | |
download | ComputeLibrary-d7e2ec51239e2075f931e0a9364e0a68534676f1.tar.gz |
Decouple NEInstanceNormalizationLayerKernel
Resolves COMPMID-4620
Signed-off-by: Dana Zlotnik <dana.zlotnik@arm.com>
Change-Id: I22c285339840493c9cfd4c1abfbc3768ad4db824
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6871
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Giorgio Arena <giorgio.arena@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h')
-rw-r--r-- | src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h | 15 |
1 files changed, 7 insertions, 8 deletions
diff --git a/src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h b/src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h index 96c0119719..f166ce2058 100644 --- a/src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h +++ b/src/core/NEON/kernels/NEInstanceNormalizationLayerKernel.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2019-2020 Arm Limited. + * Copyright (c) 2019-2022 Arm Limited. * * SPDX-License-Identifier: MIT * @@ -84,13 +84,12 @@ private: */ using NormalizationFunction = void(ITensor *input, ITensor *output, float gamma, float beta, float epsilon, const Window &window); - NormalizationFunction *_func; - ITensor *_input; - ITensor *_output; - float _gamma; - float _beta; - float _epsilon; - bool _use_mixed_precision{ true }; + ITensor *_input; + ITensor *_output; + float _gamma; + float _beta; + float _epsilon; + bool _use_mixed_precision{ true }; }; } // namespace arm_compute #endif /*ARM_COMPUTE_NEINSTANCENORMALIZATIONLAYERKERNEL_H */ |