diff options
Diffstat (limited to 'src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp')
| -rw-r--r-- | src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp b/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp index 7500b5a3ac..06d243a6d1 100644 --- a/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp +++ b/src/backends/cl/workloads/ClSpaceToBatchNdWorkload.hpp @@ -22,7 +22,9 @@ arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo& input, class ClSpaceToBatchNdWorkload : public BaseWorkload<SpaceToBatchNdQueueDescriptor> { public: - ClSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor& descriptor, const WorkloadInfo& info); + ClSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor& descriptor, + const WorkloadInfo& info, + const arm_compute::CLCompileContext& clCompileContext); void Execute() const override; private: |