aboutsummaryrefslogtreecommitdiff
path: root/src/armnn/backends/RefWorkloads/BatchNormImpl.hpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnn/backends/RefWorkloads/BatchNormImpl.hpp')
-rw-r--r--src/armnn/backends/RefWorkloads/BatchNormImpl.hpp56
1 files changed, 0 insertions, 56 deletions
diff --git a/src/armnn/backends/RefWorkloads/BatchNormImpl.hpp b/src/armnn/backends/RefWorkloads/BatchNormImpl.hpp
deleted file mode 100644
index a7579c8373..0000000000
--- a/src/armnn/backends/RefWorkloads/BatchNormImpl.hpp
+++ /dev/null
@@ -1,56 +0,0 @@
-//
-// Copyright © 2017 Arm Ltd. All rights reserved.
-// SPDX-License-Identifier: MIT
-//
-
-#pragma once
-
-#include "RefWorkloadUtils.hpp"
-
-#include <armnn/Tensor.hpp>
-
-#include <cmath>
-
-namespace armnn
-{
-
-template<typename NormData>
-static void BatchNormImpl(NormData data,
- const float* varIn,
- const float* meanIn,
- const float* gammaIn,
- const float* betaIn,
- float * outputData,
- const float * inputData)
-{
- const TensorInfo& inputInfo0 = GetTensorInfo(data.m_Inputs[0]);
- for (unsigned int c = 0; c < inputInfo0.GetShape()[1]; c++)
- {
- float var = varIn[c];
- float mean = meanIn[c];
- float gamma = gammaIn[c];
- float beta = betaIn[c];
-
- float mult = gamma / sqrtf(var + data.m_Parameters.m_Eps);
- float add = beta - mult * mean;
-
- for (unsigned int n = 0; n < inputInfo0.GetShape()[0]; n++)
- {
- for (unsigned int j = 0; j < inputInfo0.GetShape()[2]; j++)
- {
- for (unsigned int i = 0; i < inputInfo0.GetShape()[3]; i++)
- {
- unsigned int index = i +
- j*inputInfo0.GetShape()[3] +
- c*inputInfo0.GetShape()[3] * inputInfo0.GetShape()[2] +
- n*inputInfo0.GetShape()[3] * inputInfo0.GetShape()[2]
- * inputInfo0.GetShape()[1];
-
- outputData[index] = mult * inputData[index] + add;
- }
- }
- }
- }
-}
-
-} //namespace armnn