diff options
author | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-03-02 09:41:13 +0000 |
---|---|---|
committer | Sang-Hoon Park <sang-hoon.park@arm.com> | 2021-03-03 15:44:15 +0000 |
commit | 5ff38da7e18e91243a7f6b8e642f8b40f5846068 (patch) | |
tree | f0e1ef0fa2ddeca0e173de33d6d526387d5a4185 /src/runtime/CL/functions/CLPReluLayer.cpp | |
parent | 473cb01e84cef6cab057e9492bfa3b68f708e5d7 (diff) | |
download | ComputeLibrary-5ff38da7e18e91243a7f6b8e642f8b40f5846068.tar.gz |
Create ClPRelu operator
Make the class that was in experimental namespace
as ClOperator to prepare porting to new interface.
The followings are added as a part of this change
Also, in-place computation is now correctly considered
to be aligned with the class description. Test cases
to test in-place computation are added.
Partially Implements: COMPMID-4184
Signed-off-by: Sang-Hoon Park <sang-hoon.park@arm.com>
Change-Id: I71c18ab47fe0370a2060d5303a58ff3650c0093f
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5201
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'src/runtime/CL/functions/CLPReluLayer.cpp')
-rw-r--r-- | src/runtime/CL/functions/CLPReluLayer.cpp | 46 |
1 files changed, 12 insertions, 34 deletions
diff --git a/src/runtime/CL/functions/CLPReluLayer.cpp b/src/runtime/CL/functions/CLPReluLayer.cpp index 74286d46ca..bb7aff218d 100644 --- a/src/runtime/CL/functions/CLPReluLayer.cpp +++ b/src/runtime/CL/functions/CLPReluLayer.cpp @@ -21,44 +21,22 @@ * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE * SOFTWARE. */ -#include "src/core/gpu/cl/kernels/ClElementwiseKernel.h" - -#include "arm_compute/core/CL/ICLTensor.h" -#include "arm_compute/runtime/CL/CLScheduler.h" #include "arm_compute/runtime/CL/functions/CLPReluLayer.h" +#include "arm_compute/core/CL/CLKernelLibrary.h" +#include "arm_compute/core/CL/ICLTensor.h" +#include "src/core/gpu/cl/IClKernel.h" +#include "src/runtime/gpu/cl/operators/ClPRelu.h" namespace arm_compute { -namespace experimental -{ -CLPReluLayer::CLPReluLayer() -{ -} - -void CLPReluLayer::configure(const CLCompileContext &compile_context, ITensorInfo *input, ITensorInfo *alpha, ITensorInfo *output) -{ - auto k = std::make_unique<arm_compute::opencl::kernels::ClArithmeticKernel>(); - k->configure(compile_context, ArithmeticOperation::PRELU, input, alpha, output); - _kernel = std::move(k); -} - -Status CLPReluLayer::validate(const ITensorInfo *input, const ITensorInfo *alpha, const ITensorInfo *output) -{ - return arm_compute::opencl::kernels::ClArithmeticKernel::validate(ArithmeticOperation::PRELU, input, alpha, output); -} - -void CLPReluLayer::run(ITensorPack &tensors) -{ - ICLOperator::run(tensors); -} -} // namespace experimental +using OperatorType = opencl::ClPRelu; struct CLPReluLayer::Impl { - const ICLTensor *src_0{ nullptr }; - const ICLTensor *src_1{ nullptr }; - ICLTensor *dst{ nullptr }; - std::unique_ptr<experimental::CLPReluLayer> op{ nullptr }; + const ICLTensor *src_0{ nullptr }; + const ICLTensor *src_1{ nullptr }; + ICLTensor *dst{ nullptr }; + std::unique_ptr<OperatorType> op{ nullptr }; }; CLPReluLayer::CLPReluLayer() @@ -79,13 +57,13 @@ void CLPReluLayer::configure(const CLCompileContext &compile_context, ICLTensor _impl->src_0 = input; _impl->src_1 = alpha; _impl->dst = output; - _impl->op = std::make_unique<experimental::CLPReluLayer>(); - _impl->op->configure(compile_context, input->info(), alpha->info(), output->info()); + _impl->op = std::make_unique<OperatorType>(); + _impl->op->configure(compile_context, input->info(), alpha->info(), (output == nullptr ? input->info() : output->info())); } Status CLPReluLayer::validate(const ITensorInfo *input, const ITensorInfo *alpha, const ITensorInfo *output) { - return experimental::CLPReluLayer::validate(input, alpha, output); + return OperatorType::validate(input, alpha, output); } void CLPReluLayer::run() |