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