diff options
Diffstat (limited to 'src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp')
-rw-r--r-- | src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp | 5 |
1 files changed, 3 insertions, 2 deletions
diff --git a/src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp b/src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp index 96596b9d9c..5dfab9cbbd 100644 --- a/src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp +++ b/src/armnn/backends/ClWorkloads/ClFullyConnectedFloat32Workload.cpp @@ -13,8 +13,9 @@ namespace armnn using namespace armcomputetensorutils; ClFullyConnectedFloat32Workload::ClFullyConnectedFloat32Workload(const FullyConnectedQueueDescriptor& descriptor, - const WorkloadInfo& info) + const WorkloadInfo& info, std::shared_ptr<arm_compute::MemoryManagerOnDemand>& memoryManager) : Float32Workload<FullyConnectedQueueDescriptor>(descriptor, info) + , m_FullyConnected(memoryManager) { BuildArmComputeTensor(m_WeightsTensor, m_Data.m_Weight->GetTensorInfo()); @@ -49,4 +50,4 @@ void ClFullyConnectedFloat32Workload::Execute() const m_FullyConnected.run(); } -} //namespace armnn
\ No newline at end of file +} //namespace armnn |