25.11
Loading...
Searching...
No Matches
Fill.cpp
Go to the documentation of this file.
1
//
2
// Copyright © 2020 Arm Ltd and Contributors. All rights reserved.
3
// SPDX-License-Identifier: MIT
4
//
5
6
#include "
Fill.hpp
"
7
8
#include "
RefWorkloadUtils.hpp
"
9
10
namespace
armnn
11
{
12
13
void
Fill
(
Encoder<float>
& output,
14
const
TensorShape
& desiredOutputShape,
15
const
float
value)
16
{
17
for
(
unsigned
int
i = 0; i < desiredOutputShape.
GetNumElements
(); ++i)
18
{
19
output[i];
20
output.
Set
(value);
21
}
22
}
23
24
}
//namespace armnn
Fill.hpp
RefWorkloadUtils.hpp
armnn::Encoder
Definition
BaseIterator.hpp:51
armnn::Encoder::Set
virtual void Set(IType right)=0
armnn::TensorShape
Definition
Tensor.hpp:21
armnn::TensorShape::GetNumElements
unsigned int GetNumElements() const
Function that calculates the tensor elements by multiplying all dimension size which are Specified.
Definition
Tensor.cpp:181
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition
01_00_quick_start.dox:7
armnn::LayerType::Fill
@ Fill
Definition
Types.hpp:496
src
backends
reference
workloads
Fill.cpp
Generated on
for Arm NN by
1.14.0