diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2021-02-25 20:15:01 +0000 |
---|---|---|
committer | Georgios Pinitas <georgios.pinitas@arm.com> | 2021-03-29 20:23:11 +0000 |
commit | 2788609b8a10306e9eae47543b39812a7b075aaa (patch) | |
tree | 81515046e0c06d6a21ecdcebfe083ea5922fea0c /src/core/gpu/cl/kernels/ClPermuteKernel.cpp | |
parent | f9a611a1fd309bb9a906c99eede5e6b7bceba26b (diff) | |
download | ComputeLibrary-2788609b8a10306e9eae47543b39812a7b075aaa.tar.gz |
Port ClTranspose to new API
Partially Resolves: COMPMID-4277 (1/2)
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I704c2303135cbe1ba46d2fd5642c84c562204bc7
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/5194
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Georgios Pinitas <georgios.pinitas@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'src/core/gpu/cl/kernels/ClPermuteKernel.cpp')
-rw-r--r-- | src/core/gpu/cl/kernels/ClPermuteKernel.cpp | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/src/core/gpu/cl/kernels/ClPermuteKernel.cpp b/src/core/gpu/cl/kernels/ClPermuteKernel.cpp index 04e649b911..ffc13060a8 100644 --- a/src/core/gpu/cl/kernels/ClPermuteKernel.cpp +++ b/src/core/gpu/cl/kernels/ClPermuteKernel.cpp @@ -57,7 +57,7 @@ Status validate_arguments(const ITensorInfo *src, const ITensorInfo *dst, const ARM_COMPUTE_RETURN_ERROR_ON_NULLPTR(src, dst); ARM_COMPUTE_RETURN_ERROR_ON(src->data_type() == DataType::UNKNOWN); ARM_COMPUTE_RETURN_ERROR_ON_MSG(src->num_dimensions() < 1 || src->num_dimensions() > 4, - "Permutation upto 4-D src tensor is supported"); + "Permutation up to 4-D src tensor is supported"); ARM_COMPUTE_RETURN_ERROR_ON_MSG(perm.num_dimensions() < 1 || perm.num_dimensions() > 4, "Permutation vector size should be less than or equal to 4"); for(const auto &p : perm) @@ -77,17 +77,12 @@ Status validate_arguments(const ITensorInfo *src, const ITensorInfo *dst, const } } // namespace -void ClPermuteKernel::configure(const ITensorInfo *src, ITensorInfo *dst, const PermutationVector &perm) -{ - configure(CLKernelLibrary::get().get_compile_context(), src, dst, perm); -} - void ClPermuteKernel::configure(const CLCompileContext &compile_context, const ITensorInfo *src, ITensorInfo *dst, const PermutationVector &perm) { ARM_COMPUTE_ERROR_ON_NULLPTR(src, dst); auto padding_info = get_padding_info({ src, dst }); const TensorShape dst_shape = get_dst_shape(src, perm); - // Output auto inizialitation if not yet initialized + // Output auto initialization if not yet initialized auto_init_if_empty(*dst, src->clone()->set_tensor_shape(dst_shape)); ARM_COMPUTE_ERROR_THROW_ON(validate_arguments(src, dst, perm)); |