24.04
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_INDIRECT_CONV2D_H
25 #define ARM_COMPUTE_CL_INDIRECT_CONV2D_H
41 struct DirectConvComputeKernelInfo;
112 std::unique_ptr<IClKernel> _indirect_conv_kernel{
nullptr};
113 std::unique_ptr<IClKernel> _addr_precalculation_kernel{
nullptr};
115 bool _is_prepared{
false};
std::vector< MemoryInfo > MemoryRequirements
SimpleTensor< float > src
ClIndirectConv2d()=default
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Static function to check if given info will lead to a valid configuration.
Activation Layer Information class.
Basic function to execute indirect convolution on OpenCL.
Basic interface for functions which have a single async CL kernel.
void prepare(ITensorPack &constants) override
Prepare the function for executing.
experimental::MemoryRequirements workspace() const override
Return the memory requirements required by the workspace.
void configure(const CLCompileContext &compile_context, ITensorInfo *src, ITensorInfo *weights, ITensorInfo *biases, ITensorInfo *dst, const PadStrideInfo &conv_info, const ActivationLayerInfo &act_info=ActivationLayerInfo())
Initialise the kernel's inputs and output.
Store the tensor's metadata.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.