24.02
|
Go to the documentation of this file.
35 struct NEPooling3dLayer::Impl
37 const ITensor *
src{
nullptr};
38 ITensor *
dst{
nullptr};
39 std::unique_ptr<cpu::CpuPool3d> op{
nullptr};
40 MemoryGroup memory_group{};
41 ITensorPack run_pack{};
42 WorkspaceData<Tensor> workspace_tensors{};
49 _impl->memory_group =
MemoryGroup(std::move(memory_manager));
56 _impl->op = std::make_unique<cpu::CpuPool3d>();
57 _impl->op->configure(
input->info(), output->
info(), pool_info);
60 _impl->workspace_tensors = manage_workspace<Tensor>(_impl->op->workspace(), _impl->memory_group, _impl->run_pack);
73 _impl->op->run(_impl->run_pack);
SimpleTensor< float > src
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const Pooling3dLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration of NEPooling3dLayer.
void run() override
Run the kernels contained in the function.
Interface for CPU tensor.
#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.
~NEPooling3dLayer()
Default destructor.
static Status validate(const ITensorInfo *src, const ITensorInfo *dst, const Pooling3dLayerInfo &pool_info)
Static function to check if given info will lead to a valid configuration.
NEPooling3dLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr)
Constructor.
Pooling Layer Information struct.
void configure(const ITensor *input, ITensor *output, const Pooling3dLayerInfo &pool_info)
Set the input and output tensors.
Memory group resources scope handling class.
Copyright (c) 2017-2024 Arm Limited.
Store the tensor's metadata.