9 #include <arm_compute/runtime/CL/functions/CLBatchToSpaceLayer.h> 15 const TensorInfo& output,
16 const BatchToSpaceNdDescriptor& desc);
23 const arm_compute::CLCompileContext& clCompileContext);
29 mutable arm_compute::CLBatchToSpaceLayer m_Layer;
ClBatchToSpaceNdWorkload(const BatchToSpaceNdQueueDescriptor &descriptor, const WorkloadInfo &info, const arm_compute::CLCompileContext &clCompileContext)
arm_compute::Status ClBatchToSpaceNdWorkloadValidate(const TensorInfo &input, const TensorInfo &output, const BatchToSpaceNdDescriptor &desc)
Copyright (c) 2021 ARM Limited and Contributors.
void Execute() const override
Contains information about inputs and outputs to a layer.