13 #include <arm_compute/runtime/CL/functions/CLSpaceToBatchLayer.h> 19 const TensorInfo& output,
20 const SpaceToBatchNdDescriptor& descriptor);
27 const arm_compute::CLCompileContext& clCompileContext);
31 mutable arm_compute::CLSpaceToBatchLayer m_SpaceToBatchLayer;
ClSpaceToBatchNdWorkload(const SpaceToBatchNdQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
Copyright (c) 2021 ARM Limited and Contributors.
arm_compute::Status ClSpaceToBatchNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const SpaceToBatchNdDescriptor &descriptor)
void Execute() const override
Contains information about TensorInfos of a layer.