24.02.1
|
Go to the documentation of this file.
24 #ifndef ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H
25 #define ARM_COMPUTE_NEFULLYCONNECTEDLAYER_H
38 namespace weights_transformations
61 uint32_t
uid()
override
72 static constexpr uint32_t _uid = 0x0;
174 std::unique_ptr<Impl> _impl;
Convolution Layer Weights Information class.
Base class for all functions.
Interface for CPU tensor.
Basic function to compute a Fully Connected layer.
Fully connected layer info.
void prepare() override
Prepare the function for executing.
WeightFormat
Memory layouts for the weights tensor.
void configure(const ITensor *input, ITensor *output)
Initialise the kernel's inputs and output.
TensorAllocator * allocator()
Return a pointer to the tensor's allocator.
void configure(const ITensor *input, const ITensor *weights, const ITensor *biases, ITensor *output, FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), const WeightsInfo &weights_info=WeightsInfo())
Set the input and output tensors.
NEFullyConnectedLayer(std::shared_ptr< IMemoryManager > memory_manager=nullptr, IWeightsManager *weights_manager=nullptr)
Constructor.
~NEFullyConnectedLayer()
Default destructor.
void run() override
Run the kernels contained in the function.
static Status has_opt_impl(arm_compute::WeightFormat &expected_weight_format, const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, const FullyConnectedLayerInfo &fc_info, const WeightsInfo &weights_info)
Static function that queries whether fixed-format kernel exists for a given problem description.
void allocate() override
Allocate size specified by TensorInfo of CPU memory.
Copyright (c) 2017-2024 Arm Limited.
NEFullyConnectedLayer & operator=(const NEFullyConnectedLayer &)=delete
Prevent instances of this class from being copied (As this class contains pointers)
void run() override
Run the kernels contained in the function.
void free() override
Free allocated CPU memory.
Store the tensor's metadata.
Basic implementation of the tensor interface.
static Status validate(const ITensorInfo *input, const ITensorInfo *weights, const ITensorInfo *biases, const ITensorInfo *output, FullyConnectedLayerInfo fc_info=FullyConnectedLayerInfo(), const WeightsInfo &weights_info=WeightsInfo())
Static function to check if given info will lead to a valid configuration of NEFullyConnectedLayer.
Weights manager interface to handle weights transformations.
Basic function to run cpu::kernels::CpuTransposeKernel.