aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp
diff options
context:
space:
mode:
authorSadik Armagan <sadik.armagan@arm.com>2020-12-02 11:28:58 +0000
committerSadik Armagan <sadik.armagan@arm.com>2020-12-02 11:29:35 +0000
commite9444751017fe108ce80fd5c270d04fffeb14e1e (patch)
tree29601fdf9017a2dbce71983b995709c24e6c773c /src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp
parentbbbefecd34a9420bcb003dd230402c55ee5150d5 (diff)
downloadarmnn-e9444751017fe108ce80fd5c270d04fffeb14e1e.tar.gz
IVGCVSW-5482 'Add a ClCompileContext parameter to each ClWorkload Constructor'
* Injected CLCompileContext object to each CL workload. Signed-off-by: Sadik Armagan <sadik.armagan@arm.com> Change-Id: I4837dbd3d5b56cf743b3b89c944e3cdf8b11a42a
Diffstat (limited to 'src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp')
-rw-r--r--src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp7
1 files changed, 5 insertions, 2 deletions
diff --git a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp b/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp
index 443c56b7b5..7b29cded0f 100644
--- a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp
+++ b/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.cpp
@@ -45,7 +45,9 @@ arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo& input,
}
ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload(
- const SpaceToBatchNdQueueDescriptor& descriptor, const WorkloadInfo& info)
+ const SpaceToBatchNdQueueDescriptor& descriptor,
+ const WorkloadInfo& info,
+ const arm_compute::CLCompileContext& clCompileContext)
: BaseWorkload<SpaceToBatchNdQueueDescriptor>(descriptor, info)
{
m_Data.ValidateInputsOutputs("ClSpaceToBatchNdWorkload", 1, 1);
@@ -68,7 +70,8 @@ ClSpaceToBatchNdWorkload::ClSpaceToBatchNdWorkload(
input.info()->set_data_layout(aclDataLayout);
output.info()->set_data_layout(aclDataLayout);
- m_SpaceToBatchLayer.configure(&input,
+ m_SpaceToBatchLayer.configure(clCompileContext,
+ &input,
blockWidth,
blockHeight,
paddingLeftTop,