12 #include <arm_compute/runtime/CL/functions/CLDepthToSpaceLayer.h> 18 const TensorInfo& output,
26 const arm_compute::CLCompileContext& clCompileContext);
30 mutable arm_compute::CLDepthToSpaceLayer m_Layer;
Copyright (c) 2021 ARM Limited and Contributors.
ClDepthToSpaceWorkload(const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Contains information about inputs and outputs to a layer.
void Execute() const override
arm_compute::Status ClDepthToSpaceWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &desc)
SpaceToDepthDescriptor DepthToSpaceDescriptor
A DepthToSpaceDescriptor for the DepthToSpaceLayer.