12 #include <arm_compute/runtime/NEON/functions/NEDepthToSpaceLayer.h> 18 const TensorInfo& output,
28 mutable arm_compute::NEDepthToSpaceLayer m_Layer;
SpaceToDepthDescriptor DepthToSpaceDescriptor
A DepthToSpaceDescriptor for the DepthToSpaceLayer.
NeonDepthToSpaceWorkload(const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info)
arm_compute::Status NeonDepthToSpaceWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor)
void Execute() const override