24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CL_DIRECT_CONV3D_H
25 #define ARM_COMPUTE_CL_DIRECT_CONV3D_H
34 class CLCompileContext;
93 std::unique_ptr<IClKernel> _direct_conv3d_kernel{
nullptr};
void run(ITensorPack &tensors) override
Run the kernels contained in the function.
arm_compute::ICLKernel IClKernel
Basic function to simulate a directly convolution layer with 3 spatial dimensions.
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo &conv3d_info)
Static function to check if given info will lead to a valid configuration.
Basic interface for functions which have a single async CL kernel.
Copyright (c) 2017-2024 Arm Limited.
Descriptor used by the 3d Convolution function.
void configure(const CLCompileContext &compile_context, const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *src2, ITensorInfo *dst, const Conv3dInfo &conv3d_info)
Set the src and dst tensors.
Store the tensor's metadata.