diff options
Diffstat (limited to 'tests/networks/MobileNetV1Network.h')
-rw-r--r-- | tests/networks/MobileNetV1Network.h | 7 |
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) { |