24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_CLCONVOLUTION3DLAYER_H
25 #define ARM_COMPUTE_CLCONVOLUTION3DLAYER_H
33 class CLCompileContext;
113 std::unique_ptr<Impl> _impl;
SimpleTensor< float > src
CLConv3D & operator=(const CLConv3D &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
Base class for all functions.
Interface for OpenCL tensor.
void run() override
Run the kernels contained in the function.
static Status validate(const ITensorInfo *src, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *dst, const Conv3dInfo &conv3d_info)
Static function to check if given info will lead to a valid configuration of CLConv3D.
void configure(const CLCompileContext &compile_context, const ICLTensor *src, const ICLTensor *weights, const ICLTensor *biases, ICLTensor *dst, const Conv3dInfo &conv3d_info)
Set the src and dst tensors.
Copyright (c) 2017-2024 Arm Limited.
Descriptor used by the 3d Convolution function.
Store the tensor's metadata.
Basic function to compute the convolution3d layer.