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