24.02.1
|
Go to the documentation of this file.
45 auto kernel_object = std::make_unique<kernels::ClTransposedConvolutionKernel>();
47 kernel_object->configure(compile_context,
input, weights, biases, output, deconv_info);
48 _transposed_conv_kernel = std::move(kernel_object);
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
void configure(const CLCompileContext &compile_context, const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, ITensorInfo *output, const PadStrideInfo &deconv_info)
Set the input, weights, biases and output tensors.
Interface to enqueue OpenCL kernels and get/set the OpenCL CommandQueue and ICLTuner.
static CLScheduler & get()
Access the scheduler singleton.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &deconv_info)
Static function to check if given info will lead to a valid configuration.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const PadStrideInfo &deconv_info)
Static function to check if given info will lead to a valid configuration.
Copyright (c) 2017-2024 Arm Limited.
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
void enqueue_op(ICLKernel &kernel, ITensorPack &tensors, bool flush=true)
Schedule the execution of the passed kernel if possible.
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)