diff options
author | Teresa Charlin <teresa.charlinreyes@arm.com> | 2023-07-18 12:56:19 +0100 |
---|---|---|
committer | mike.kelly <mike.kelly@arm.com> | 2023-07-28 19:14:05 +0000 |
commit | 4f9a6d63cd263789377b97cc409e0c4c6d707830 (patch) | |
tree | 800edbe13c4e02a5169b9e55ed3d12622413c4db /src/backends/neon/workloads | |
parent | a0cc6ff8065083b833dda9c3c5992938834bdf29 (diff) | |
download | armnn-4f9a6d63cd263789377b97cc409e0c4c6d707830.tar.gz |
Add datalayout to tensors in NeonStridedSliceWorkload
Signed-off-by: Teresa Charlin <teresa.charlinreyes@arm.com>
Change-Id: I9464970b0aa363f6ac3e0cb42ccb2bb3c8724ca4
Diffstat (limited to 'src/backends/neon/workloads')
-rw-r--r-- | src/backends/neon/workloads/NeonStridedSliceWorkload.cpp | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/src/backends/neon/workloads/NeonStridedSliceWorkload.cpp b/src/backends/neon/workloads/NeonStridedSliceWorkload.cpp index 8a5194720f..219c40c302 100644 --- a/src/backends/neon/workloads/NeonStridedSliceWorkload.cpp +++ b/src/backends/neon/workloads/NeonStridedSliceWorkload.cpp @@ -20,8 +20,10 @@ arm_compute::Status NeonStridedSliceWorkloadValidate(const TensorInfo& input, const TensorInfo& output, const StridedSliceDescriptor& descriptor) { - const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input); - const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output); + const arm_compute::TensorInfo aclInput = armcomputetensorutils::BuildArmComputeTensorInfo(input, + descriptor.m_DataLayout); + const arm_compute::TensorInfo aclOutput = armcomputetensorutils::BuildArmComputeTensorInfo(output, + descriptor.m_DataLayout); arm_compute::Coordinates starts; arm_compute::Coordinates ends; |