diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-08-16 12:38:54 +0100 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-08-20 16:26:16 +0000 |
commit | 19884630c37ae9de2f65a88ea2cda5630a551bad (patch) | |
tree | 0fdffe84c5d66ffe8a1d320b798a247afa1dfdae /src/runtime/gpu/cl/operators/ClConv2d.cpp | |
parent | 73df9310f4e94e43597f283307e3cde0653d8bae (diff) | |
download | ComputeLibrary-19884630c37ae9de2f65a88ea2cda5630a551bad.tar.gz |
Rename [Cl|Cpu]GemmConvolution to [Cl|Gpu]GemmConv2d
Renaming the gemm-based convolution operators to accomodate for new
operators with higher convolution dimensonality
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Id2f2cf11404221f0e87baa0e5d08ad5d63eaf78e
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/6113
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/runtime/gpu/cl/operators/ClConv2d.cpp')
-rw-r--r-- | src/runtime/gpu/cl/operators/ClConv2d.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/src/runtime/gpu/cl/operators/ClConv2d.cpp b/src/runtime/gpu/cl/operators/ClConv2d.cpp index 2f4d673d9c..0cb3a968e6 100644 --- a/src/runtime/gpu/cl/operators/ClConv2d.cpp +++ b/src/runtime/gpu/cl/operators/ClConv2d.cpp @@ -31,7 +31,7 @@ #include "arm_compute/runtime/CL/CLScheduler.h" #include "arm_compute/runtime/CL/functions/CLFFTConvolutionLayer.h" #include "src/runtime/gpu/cl/operators/ClDirectConv2d.h" -#include "src/runtime/gpu/cl/operators/ClGemmConvolution.h" +#include "src/runtime/gpu/cl/operators/ClGemmConv2d.h" #include "src/runtime/gpu/cl/operators/ClWinogradConv2d.h" #include <memory> @@ -104,7 +104,7 @@ void ClConv2d::configure(const CLCompileContext &compile_context, ITensorInfo *s } case ConvolutionMethod::GEMM: { - auto f = std::make_unique<ClGemmConvolution>(); + auto f = std::make_unique<ClGemmConv2d>(); f->configure(compile_context, src, weights, biases, dst, conv2d_info, weights_info); _operator = std::move(f); break; @@ -143,7 +143,7 @@ Status ClConv2d::validate(const ITensorInfo *src, const ITensorInfo *weights, co case ConvolutionMethod::GEMM: { // Validate gemm-based convolution layer - ARM_COMPUTE_RETURN_ON_ERROR(ClGemmConvolution::validate(src, weights, biases, dst, conv2d_info, weights_info)); + ARM_COMPUTE_RETURN_ON_ERROR(ClGemmConv2d::validate(src, weights, biases, dst, conv2d_info, weights_info)); break; } default: |