aboutsummaryrefslogtreecommitdiff
path: root/tests/networks/MobileNetV1Network.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/networks/MobileNetV1Network.h')
-rw-r--r--tests/networks/MobileNetV1Network.h7
1 files changed, 7 insertions, 0 deletions
diff --git a/tests/networks/MobileNetV1Network.h b/tests/networks/MobileNetV1Network.h
index dbe3f491f0..0957c6b555 100644
--- a/tests/networks/MobileNetV1Network.h
+++ b/tests/networks/MobileNetV1Network.h
@@ -300,6 +300,13 @@ public:
smx.run();
}
+ /** Sync the results */
+ void sync()
+ {
+ sync_if_necessary<TensorType>();
+ sync_tensor_if_necessary<TensorType>(output);
+ }
+
private:
void depthwise_conv_block_init(unsigned int idx, unsigned int ifm, unsigned int ofm)
{