aboutsummaryrefslogtreecommitdiff
path: root/tests/benchmark/fixtures/DequantizationLayerFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/benchmark/fixtures/DequantizationLayerFixture.h')
-rw-r--r--tests/benchmark/fixtures/DequantizationLayerFixture.h6
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/benchmark/fixtures/DequantizationLayerFixture.h b/tests/benchmark/fixtures/DequantizationLayerFixture.h
index 30e05f9c3b..4251a96881 100644
--- a/tests/benchmark/fixtures/DequantizationLayerFixture.h
+++ b/tests/benchmark/fixtures/DequantizationLayerFixture.h
@@ -73,6 +73,12 @@ public:
dequantization_func.run();
}
+ void sync()
+ {
+ sync_if_necessary<TensorType>();
+ sync_tensor_if_necessary<TensorType>(dst);
+ }
+
void teardown()
{
src.allocator()->free();