12 #include <arm_compute/runtime/NEON/functions/NEDepthToSpaceLayer.h> 18 const TensorInfo& output,
28 mutable arm_compute::NEDepthToSpaceLayer m_Layer;
NeonDepthToSpaceWorkload(const DepthToSpaceQueueDescriptor &descriptor, const WorkloadInfo &info)
Copyright (c) 2020 ARM Limited.
Contains information about inputs and outputs to a layer.
arm_compute::Status NeonDepthToSpaceWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const DepthToSpaceDescriptor &descriptor)
void Execute() const override
SpaceToDepthDescriptor DepthToSpaceDescriptor
A DepthToSpaceDescriptor for the DepthToSpaceLayer.