aboutsummaryrefslogtreecommitdiff
path: root/tests/networks/MobileNetNetwork.h
diff options
context:
space:
mode:
authorAlex Gilday <alexander.gilday@arm.com>2018-03-21 13:54:09 +0000
committerAnthony Barbier <anthony.barbier@arm.com>2018-11-02 16:49:16 +0000
commitc357c47be8a3f210f9eee9a05cc13f1051b036d3 (patch)
treea88ac857150da970a0862a3479b78c616d8aa1d3 /tests/networks/MobileNetNetwork.h
parent724079d6fce3bf6a05cd6c7b4884b132b27e9e90 (diff)
downloadComputeLibrary-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/MobileNetNetwork.h')
-rw-r--r--tests/networks/MobileNetNetwork.h7
1 files changed, 6 insertions, 1 deletions
diff --git a/tests/networks/MobileNetNetwork.h b/tests/networks/MobileNetNetwork.h
index 8c3cb1fb2d..ec054b237e 100644
--- a/tests/networks/MobileNetNetwork.h
+++ b/tests/networks/MobileNetNetwork.h
@@ -1,5 +1,5 @@
/*
- * Copyright (c) 2017 ARM Limited.
+ * Copyright (c) 2017-2018 ARM Limited.
*
* SPDX-License-Identifier: MIT
*
@@ -53,6 +53,10 @@ template <typename TensorType,
class MobileNetNetwork
{
public:
+ /** Initialize the network.
+ *
+ * @param[in] batches Number of batches.
+ */
void init(int batches)
{
_batches = batches;
@@ -105,6 +109,7 @@ public:
reshape.configure(&conv_out[14], &output);
}
+ /** Allocate the network. */
void allocate()
{
input.allocator()->allocate();