aboutsummaryrefslogtreecommitdiff
path: root/src/backends/cl/workloads/ClFullyConnectedWorkload.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/ClFullyConnectedWorkload.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/ClFullyConnectedWorkload.cpp')
-rw-r--r--src/backends/cl/workloads/ClFullyConnectedWorkload.cpp14
1 files changed, 11 insertions, 3 deletions
diff --git a/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp b/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp
index eaec639f28..9135d27376 100644
--- a/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp
+++ b/src/backends/cl/workloads/ClFullyConnectedWorkload.cpp
@@ -45,8 +45,11 @@ arm_compute::Status ClFullyConnectedWorkloadValidate(const TensorInfo& input,
fullyConnectedLayerInfo);
}
-ClFullyConnectedWorkload::ClFullyConnectedWorkload(const FullyConnectedQueueDescriptor& descriptor,
- const WorkloadInfo& info, std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager)
+ClFullyConnectedWorkload::ClFullyConnectedWorkload(
+ const FullyConnectedQueueDescriptor& descriptor,
+ const WorkloadInfo& info,
+ std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager,
+ const arm_compute::CLCompileContext& clCompileContext)
: BaseWorkload<FullyConnectedQueueDescriptor>(descriptor, info)
, m_FullyConnectedLayer(memoryManager)
{
@@ -69,7 +72,12 @@ ClFullyConnectedWorkload::ClFullyConnectedWorkload(const FullyConnectedQueueDesc
arm_compute::FullyConnectedLayerInfo fc_info =
ConvertFullyConnectedDescriptorToAclFullyConnectedLayerInfo(descriptor.m_Parameters, activationInfo);
- m_FullyConnectedLayer.configure(&input, m_WeightsTensor.get(), m_BiasesTensor.get(), &output, fc_info);
+ m_FullyConnectedLayer.configure(clCompileContext,
+ &input,
+ m_WeightsTensor.get(),
+ m_BiasesTensor.get(),
+ &output,
+ fc_info);
InitializeArmComputeClTensorData(*m_WeightsTensor, m_Data.m_Weight);