diff options
| author | Alex Gilday <alexander.gilday@arm.com> | 2018-03-21 13:54:09 +0000 |
|---|---|---|
| committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:49:16 +0000 |
| commit | c357c47be8a3f210f9eee9a05cc13f1051b036d3 (patch) | |
| tree | a88ac857150da970a0862a3479b78c616d8aa1d3 /tests/networks/AlexNetNetwork.h | |
| parent | 724079d6fce3bf6a05cd6c7b4884b132b27e9e90 (diff) | |
| download | ComputeLibrary-c357c47be8a3f210f9eee9a05cc13f1051b036d3.tar.gz | |
COMPMID-1008: Fix Doxygen issues
Change-Id: Ie73d8771f85d1f5b059f3a56f1bbd73c98e94a38
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/124723
Reviewed-by: Michalis Spyrou <michalis.spyrou@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>
Diffstat (limited to 'tests/networks/AlexNetNetwork.h')
| -rw-r--r-- | tests/networks/AlexNetNetwork.h | 11 |
1 files changed, 10 insertions, 1 deletions
diff --git a/tests/networks/AlexNetNetwork.h b/tests/networks/AlexNetNetwork.h index 819111f897..a30b7f8f75 100644 --- a/tests/networks/AlexNetNetwork.h +++ b/tests/networks/AlexNetNetwork.h @@ -1,5 +1,5 @@ /* - * Copyright (c) 2017 ARM Limited. + * Copyright (c) 2017-2018 ARM Limited. * * SPDX-License-Identifier: MIT * @@ -54,6 +54,13 @@ template <typename ITensorType, class AlexNetNetwork { public: + /** Initialize the network. + * + * @param[in] data_type Data type. + * @param[in] fixed_point_position Fixed point position (for quantized data types). + * @param[in] batches Number of batches. + * @param[in] reshaped_weights Whether the weights need reshaping or not. Default: false. + */ void init(DataType data_type, int fixed_point_position, int batches, bool reshaped_weights = false) { _data_type = data_type; @@ -185,6 +192,7 @@ public: } } + /** Build the network */ void build() { input.allocator()->init(TensorInfo(TensorShape(227U, 227U, 3U, _batches), 1, _data_type, _fixed_point_position)); @@ -270,6 +278,7 @@ public: smx.configure(&fc8_out, &output); } + /** Allocate the network */ void allocate() { input.allocator()->allocate(); |