diff options
Diffstat (limited to 'tests/benchmark/fixtures/DepthConcatenateLayerFixture.h')
-rw-r--r-- | tests/benchmark/fixtures/DepthConcatenateLayerFixture.h | 6 |
1 files changed, 6 insertions, 0 deletions
diff --git a/tests/benchmark/fixtures/DepthConcatenateLayerFixture.h b/tests/benchmark/fixtures/DepthConcatenateLayerFixture.h index 221f34c6fd..523a41b8be 100644 --- a/tests/benchmark/fixtures/DepthConcatenateLayerFixture.h +++ b/tests/benchmark/fixtures/DepthConcatenateLayerFixture.h @@ -123,6 +123,12 @@ public: _depth_concat.run(); } + void sync() + { + sync_if_necessary<TensorType>(); + sync_tensor_if_necessary<TensorType>(_dst); + } + void teardown() { for(auto &src : _srcs) |