24.02
|
Go to the documentation of this file.
39 std::unique_ptr<cpu::ICpuOperator> op{
nullptr};
40 ITensorPack run_pack{};
58 auto f = std::make_unique<cpu::CpuDirectConv3d>();
59 f->configure(
input->info(), weights->
info(), ((biases !=
nullptr) ? biases->
info() :
nullptr), output->
info(),
61 _impl->op = std::move(f);
63 if (_impl->op !=
nullptr)
82 if (_impl->op !=
nullptr)
84 _impl->op->run(_impl->run_pack);
static Status validate(const ITensorInfo *src0, const ITensorInfo *src1, const ITensorInfo *src2, const ITensorInfo *dst, const Conv3dInfo conv_info)
Static function to check if given info will lead to a valid configuration.
Interface for CPU tensor.
#define ARM_COMPUTE_RETURN_ON_ERROR(status)
Checks if a status contains an error and returns it.
#define ARM_COMPUTE_ERROR_ON_NULLPTR(...)
virtual ITensorInfo * info() const =0
Interface to be implemented by the child class to return the tensor's metadata.
void run() override
Run the kernels contained in the function.
#define ARM_COMPUTE_ERROR_THROW_ON(status)
~NEConv3D()
Default destructor.
void configure(ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, const Conv3dInfo &conv_info)
Set the input and output tensors.
Copyright (c) 2017-2024 Arm Limited.
Descriptor used by the 3d Convolution function.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const Conv3dInfo &conv_info)
Static function to check if given info will lead to a valid configuration.
Store the tensor's metadata.
#define ARM_COMPUTE_LOG_PARAMS(...)