21.02
|
#include "arm_compute/runtime/IFunction.h"
#include "arm_compute/runtime/NEON/INESimpleFunctionNoBorder.h"
#include "arm_compute/runtime/MemoryGroup.h"
#include "arm_compute/runtime/NEON/functions/NEConvertFullyConnectedWeights.h"
#include "arm_compute/runtime/NEON/functions/NEFlattenLayer.h"
#include "arm_compute/runtime/NEON/functions/NEGEMM.h"
#include "arm_compute/runtime/NEON/functions/NEGEMMLowpMatrixMultiplyCore.h"
#include "arm_compute/runtime/Tensor.h"
Go to the source code of this file.
Data Structures | |
class | NEFullyConnectedLayerReshapeWeights |
Basic function to reshape the weights of Fully Connected layer with Neon. More... | |
class | NEFullyConnectedLayerReshapeWeightsManaged |
Basic function to manage the reshape weights generated from NEFullyConnectedLayerReshapeWeights. More... | |
class | NEFullyConnectedLayer |
Basic function to compute a Fully Connected layer on Neon. More... | |
Namespaces | |
arm_compute | |
Copyright (c) 2017-2021 Arm Limited. | |
arm_compute::weights_transformations | |