24.02
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
armnn::Encoder::Set
virtual void Set(IType right)=0
armnn::Fill
void Fill(Encoder< float > &output, const TensorShape &desiredOutputShape, const float value)
Creates a tensor and fills it with a scalar value.
Definition:
Fill.cpp:13
armnn::TensorShape
Definition:
Tensor.hpp:20
armnn::Encoder< float >
Fill.hpp
RefWorkloadUtils.hpp
armnn
Copyright (c) 2021 ARM Limited and Contributors.
Definition:
01_00_quick_start.dox:6
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
src
backends
reference
workloads
Fill.cpp
Generated on Wed Feb 14 2024 16:36:17 for Arm NN by
1.8.17