diff options
author | Georgios Pinitas <georgios.pinitas@arm.com> | 2020-12-03 20:37:43 +0000 |
---|---|---|
committer | Michele Di Giorgio <michele.digiorgio@arm.com> | 2020-12-08 11:56:12 +0000 |
commit | e2696b1f9bb28b69beff99f54addd48f60823ddb (patch) | |
tree | 68705f1cdff45e4d0c174b6037f6e5ff696717d2 /arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h | |
parent | 8c3c0e7b117723bc98b6acc85565ffa521b10c0d (diff) | |
download | ComputeLibrary-e2696b1f9bb28b69beff99f54addd48f60823ddb.tar.gz |
Wrap Flatten layer over reshape
Flatten layer is lowered into a Reshape layer.
Remove (CL/NE)FlatternLayerKernel.
Partially Resolves: COMPMID-3996
Signed-off-by: Georgios Pinitas <georgios.pinitas@arm.com>
Change-Id: Id9e2ddfe2e2dd793541badff3490c05e4c908f88
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/4660
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Michele Di Giorgio <michele.digiorgio@arm.com>
Diffstat (limited to 'arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h')
-rw-r--r-- | arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h | 4 |
1 files changed, 1 insertions, 3 deletions
diff --git a/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h b/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h index 0a7748a94b..8bf3e95f81 100644 --- a/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h +++ b/arm_compute/runtime/NEON/functions/NEFullyConnectedLayer.h @@ -35,8 +35,6 @@ namespace arm_compute { -class NEFlattenLayerKernel; - /** Basic function to reshape the weights of Fully Connected layer with NEON. This function calls the following kernels: * * @note The fully connected layer accepts "weights" tensors only with 2 dimensions. @@ -181,7 +179,7 @@ private: MemoryGroup _memory_group; IWeightsManager *_weights_manager; - std::unique_ptr<NEFlattenLayerKernel> _flatten_kernel; + NEFlattenLayer _flatten; NEConvertFullyConnectedWeights _convert_weights; weights_transformations::NEConvertFullyConnectedWeightsManaged _convert_weights_managed; NEFullyConnectedLayerReshapeWeights _reshape_weights_function; |