24 #ifndef ARM_COMPUTE_NECONVERTFULLYCONNECTEDWEIGHTS_H 25 #define ARM_COMPUTE_NECONVERTFULLYCONNECTEDWEIGHTS_H 37 class NEConvertFullyConnectedWeightsKernel;
78 std::unique_ptr<NEConvertFullyConnectedWeightsKernel> _kernel;
81 namespace weights_transformations
89 _output.allocator()->allocate();
96 _output.allocator()->free();
111 _func.configure(input, &_output, original_input_shape, data_layout);
115 static constexpr uint32_t _uid = 0x4;
NEConvertFullyConnectedWeights & operator=(const NEConvertFullyConnectedWeights &)=delete
Prevent instances of this class from being copied (As this class contains pointers) ...
Base class for all functions.
const DataLayout data_layout
Store the tensor's metadata.
Interface for Neon tensor.
Copyright (c) 2017-2021 Arm Limited.
Basic function to run NEConvertFullyConnectedWeightsKernel.
NEConvertFullyConnectedWeights()
Default constructor.
Basic implementation of the tensor interface.
void configure(const ITensor *input, ITensor *output, const TensorShape &original_input_shape, DataLayout data_layout)
Initialize the function.
~NEConvertFullyConnectedWeights()
Default destructor.
static Status validate(const ITensorInfo *input, const ITensorInfo *output, const TensorShape &original_input_shape, DataLayout data_layout)
Static function to check if given info will lead to a valid configuration of NEConvertFullyConnectedW...
DataLayout
[DataLayout enum definition]
void run() override
Run the kernels contained in the function.