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