blob: cee38fc1f16565f2dff04cb2a210adf5852fea3d (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
|
//
// Copyright © 2017 Arm Ltd. All rights reserved.
// SPDX-License-Identifier: MIT
//
#include "Rsqrt.hpp"
#include <cmath>
namespace armnn
{
void Rsqrt(const float* in,
float* out,
const TensorInfo& tensorInfo)
{
for (size_t i = 0; i < tensorInfo.GetNumElements(); i++)
{
out[i] = 1.f / sqrtf(in[i]);
}
}
} //namespace armnn
|