1 //
2 // Copyright © 2022 Arm Ltd and Contributors. All rights reserved.
3 // SPDX-License-Identifier: MIT
4 //
5
6 #include "RefBatchNormalizationWorkload.hpp"
7
8 #include "BatchNormImpl.hpp"
9 #include "RefWorkloadUtils.hpp"
10
11 #include "Profiling.hpp"
12
13 namespace armnn
14 {
15
RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor & descriptor,const WorkloadInfo & info)16 RefBatchNormalizationWorkload::RefBatchNormalizationWorkload(const BatchNormalizationQueueDescriptor& descriptor,
17 const WorkloadInfo& info)
18 : RefBaseWorkload(descriptor, info)
19 , m_Mean (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Mean)))
20 , m_Variance(std::make_unique<ScopedTensorHandle>(*(descriptor.m_Variance)))
21 , m_Beta (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Beta)))
22 , m_Gamma (std::make_unique<ScopedTensorHandle>(*(descriptor.m_Gamma)))
23 {}
24
Execute() const25 void RefBatchNormalizationWorkload::Execute() const
26 {
27 Execute(m_Data.m_Inputs, m_Data.m_Outputs);
28 }
29
ExecuteAsync(ExecutionData & executionData)30 void RefBatchNormalizationWorkload::ExecuteAsync(ExecutionData& executionData)
31 {
32 WorkingMemDescriptor* workingMemDescriptor = static_cast<WorkingMemDescriptor*>(executionData.m_Data);
33 Execute(workingMemDescriptor->m_Inputs, workingMemDescriptor->m_Outputs);
34 }
35
Execute(std::vector<ITensorHandle * > inputs,std::vector<ITensorHandle * > outputs) const36 void RefBatchNormalizationWorkload::Execute(std::vector<ITensorHandle*> inputs,
37 std::vector<ITensorHandle*> outputs) const
38 {
39 ARMNN_SCOPED_PROFILING_EVENT(Compute::CpuRef, "RefBatchNormalizationWorkload_Execute");
40
41 std::unique_ptr<Decoder<float>> meanDecoder = MakeDecoder<float>(m_Mean->GetTensorInfo(),
42 m_Mean->Map(true));
43 std::unique_ptr<Decoder<float>> varianceDecoder = MakeDecoder<float>(m_Variance->GetTensorInfo(),
44 m_Variance->Map(true));
45 std::unique_ptr<Decoder<float>> gammaDecoder = MakeDecoder<float>(m_Gamma->GetTensorInfo(),
46 m_Gamma->Map(true));
47 std::unique_ptr<Decoder<float>> betaDecoder = MakeDecoder<float>(m_Beta->GetTensorInfo(),
48 m_Beta->Map(true));
49 std::unique_ptr<Decoder<float>> inputDecoder = MakeDecoder<float>(GetTensorInfo(inputs[0]),
50 inputs[0]->Map());
51 std::unique_ptr<Encoder<float>> outputEncoder = MakeEncoder<float>(GetTensorInfo(outputs[0]),
52 outputs[0]->Map());
53
54 BatchNormImpl(m_Data, *meanDecoder, *varianceDecoder, *betaDecoder, *gammaDecoder, *inputDecoder, *outputEncoder);
55 }
56
57 } // namespace armnn
58