diff options
author | Joel Liang <joel.liang@arm.com> | 2017-12-28 10:09:51 +0800 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:42:33 +0000 |
commit | 1c5ffd612979d40730feecc4f43fa6c9f177f2e3 (patch) | |
tree | 510367ba70fcad4374d2cf9cdccdb3bc715c216c /tests/networks/LeNet5Network.h | |
parent | c5a7e59655b61ad617fa34a4fb00e1a007c8255a (diff) | |
download | ComputeLibrary-1c5ffd612979d40730feecc4f43fa6c9f177f2e3.tar.gz |
APPBROWSER-359: Sync tensor in GC benchmark tests
Change-Id: I22c1aa92e70d6143bbcec90e9e7de9f1ce1c1e55
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/114635
Tested-by: Jenkins <bsgcomp@arm.com>
Reviewed-by: Pablo Tello <pablo.tello@arm.com>
Diffstat (limited to 'tests/networks/LeNet5Network.h')
-rw-r--r-- | tests/networks/LeNet5Network.h | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/networks/LeNet5Network.h b/tests/networks/LeNet5Network.h index f8aac30807..a46489f88c 100644 --- a/tests/networks/LeNet5Network.h +++ b/tests/networks/LeNet5Network.h @@ -230,6 +230,13 @@ public: smx.run(); } + /** Sync the results */ + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(output); + } + private: unsigned int _batches{ 0 }; |