aboutsummaryrefslogtreecommitdiff
path: root/src/backends/reference/workloads/Fill.cpp
blob: a166a61ea46aef65e105205635c7483833a24ca7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
//
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
// SPDX-License-Identifier: MIT
//

#include "Fill.hpp"

#include "RefWorkloadUtils.hpp"

namespace armnn
{

void Fill(Encoder<float>& output,
          const TensorShape& desiredOutputShape,
          const float value)
{
    for(unsigned int i = 0; i < desiredOutputShape.GetNumElements(); ++i)
    {
        output[i];
        output.Set(value);
    }
}

} //namespace armnn