13 #include <arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h> 19 const TensorInfo& output,
20 const SpaceToBatchNdDescriptor& descriptor);
29 mutable arm_compute::CLSpaceToBatchLayer m_SpaceToBatchLayer;
Copyright (c) 2020 ARM Limited.
arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor)
void Execute() const override
ClSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info)
Contains information about inputs and outputs to a layer.