13 #include <arm_compute/runtime/NEON/functions/NESpaceToBatchLayer.h> 19 const TensorInfo& output,
20 const SpaceToBatchNdDescriptor& descriptor);
29 virtual void Execute()
const override;
32 mutable std::unique_ptr<arm_compute::NESpaceToBatchLayer> m_Layer;
virtual void Execute() const override
Copyright (c) 2020 ARM Limited.
arm_compute::Status NeonSpaceToBatchNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor)
NeonSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about inputs and outputs to a layer.