xref: /aosp_15_r20/external/ComputeLibrary/src/runtime/NEON/functions/NEDirectConvolutionLayer.cpp (revision c217d954acce2dbc11938adb493fc0abd69584f3)
1*c217d954SCole Faust /*
2*c217d954SCole Faust  * Copyright (c) 2017-2021 Arm Limited.
3*c217d954SCole Faust  *
4*c217d954SCole Faust  * SPDX-License-Identifier: MIT
5*c217d954SCole Faust  *
6*c217d954SCole Faust  * Permission is hereby granted, free of charge, to any person obtaining a copy
7*c217d954SCole Faust  * of this software and associated documentation files (the "Software"), to
8*c217d954SCole Faust  * deal in the Software without restriction, including without limitation the
9*c217d954SCole Faust  * rights to use, copy, modify, merge, publish, distribute, sublicense, and/or
10*c217d954SCole Faust  * sell copies of the Software, and to permit persons to whom the Software is
11*c217d954SCole Faust  * furnished to do so, subject to the following conditions:
12*c217d954SCole Faust  *
13*c217d954SCole Faust  * The above copyright notice and this permission notice shall be included in all
14*c217d954SCole Faust  * copies or substantial portions of the Software.
15*c217d954SCole Faust  *
16*c217d954SCole Faust  * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
17*c217d954SCole Faust  * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
18*c217d954SCole Faust  * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
19*c217d954SCole Faust  * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
20*c217d954SCole Faust  * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
21*c217d954SCole Faust  * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
22*c217d954SCole Faust  * SOFTWARE.
23*c217d954SCole Faust  */
24*c217d954SCole Faust #include "arm_compute/runtime/NEON/functions/NEDirectConvolutionLayer.h"
25*c217d954SCole Faust 
26*c217d954SCole Faust #include "arm_compute/core/PixelValue.h"
27*c217d954SCole Faust #include "arm_compute/core/Utils.h"
28*c217d954SCole Faust #include "arm_compute/core/Validate.h"
29*c217d954SCole Faust #include "arm_compute/runtime/NEON/NEScheduler.h"
30*c217d954SCole Faust #include "src/cpu/operators/CpuDirectConv2d.h"
31*c217d954SCole Faust 
32*c217d954SCole Faust namespace arm_compute
33*c217d954SCole Faust {
34*c217d954SCole Faust struct NEDirectConvolutionLayer::Impl
35*c217d954SCole Faust {
36*c217d954SCole Faust     ITensor                              *src{ nullptr };
37*c217d954SCole Faust     const ITensor                        *weights{ nullptr };
38*c217d954SCole Faust     const ITensor                        *bias{ nullptr };
39*c217d954SCole Faust     ITensor                              *dst{ nullptr };
40*c217d954SCole Faust     std::unique_ptr<cpu::CpuDirectConv2d> op{ nullptr };
41*c217d954SCole Faust };
42*c217d954SCole Faust 
NEDirectConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager)43*c217d954SCole Faust NEDirectConvolutionLayer::NEDirectConvolutionLayer(std::shared_ptr<IMemoryManager> memory_manager)
44*c217d954SCole Faust     : _memory_manager(std::move(memory_manager)), _impl(std::make_unique<Impl>())
45*c217d954SCole Faust {
46*c217d954SCole Faust }
47*c217d954SCole Faust NEDirectConvolutionLayer::~NEDirectConvolutionLayer() = default;
48*c217d954SCole Faust 
configure(ITensor * input,const ITensor * weights,const ITensor * bias,ITensor * output,const PadStrideInfo & conv_info,const ActivationLayerInfo & act_info)49*c217d954SCole Faust void NEDirectConvolutionLayer::configure(ITensor *input, const ITensor *weights, const ITensor *bias, ITensor *output, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info)
50*c217d954SCole Faust {
51*c217d954SCole Faust     _impl->src     = input;
52*c217d954SCole Faust     _impl->weights = weights;
53*c217d954SCole Faust     _impl->bias    = bias;
54*c217d954SCole Faust     _impl->dst     = output;
55*c217d954SCole Faust     _impl->op      = std::make_unique<cpu::CpuDirectConv2d>(_memory_manager);
56*c217d954SCole Faust     _impl->op->configure(input->info(), weights->info(), (bias != nullptr ? bias->info() : nullptr), output->info(), conv_info, act_info);
57*c217d954SCole Faust }
58*c217d954SCole Faust 
validate(const ITensorInfo * input,const ITensorInfo * weights,const ITensorInfo * bias,const ITensorInfo * output,const PadStrideInfo & conv_info,const ActivationLayerInfo & act_info)59*c217d954SCole Faust Status NEDirectConvolutionLayer::validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *bias, const ITensorInfo *output, const PadStrideInfo &conv_info,
60*c217d954SCole Faust                                           const ActivationLayerInfo &act_info)
61*c217d954SCole Faust {
62*c217d954SCole Faust     return cpu::CpuDirectConv2d::validate(input, weights, bias, output, conv_info, act_info);
63*c217d954SCole Faust }
64*c217d954SCole Faust 
run()65*c217d954SCole Faust void NEDirectConvolutionLayer::run()
66*c217d954SCole Faust {
67*c217d954SCole Faust     ITensorPack pack;
68*c217d954SCole Faust     pack.add_tensor(TensorType::ACL_SRC_0, _impl->src);
69*c217d954SCole Faust     pack.add_tensor(TensorType::ACL_SRC_1, _impl->weights);
70*c217d954SCole Faust     pack.add_tensor(TensorType::ACL_SRC_2, _impl->bias);
71*c217d954SCole Faust     pack.add_tensor(TensorType::ACL_DST, _impl->dst);
72*c217d954SCole Faust     _impl->op->run(pack);
73*c217d954SCole Faust }
74*c217d954SCole Faust } // namespace arm_compute
75