diff options
author | Gian Marco Iodice <gianmarco.iodice@arm.com> | 2017-08-15 11:45:22 +0100 |
---|---|---|
committer | Anthony Barbier <anthony.barbier@arm.com> | 2018-11-02 16:35:24 +0000 |
commit | edfa9f463bed084f8b0953557202b2a1e56da817 (patch) | |
tree | 5d1e92926d112fde05dcbc61324d96f73f692390 /tests/model_objects | |
parent | dc460f13ee65e27b2a428e44c2d80afb1f516a99 (diff) | |
download | ComputeLibrary-edfa9f463bed084f8b0953557202b2a1e56da817.tar.gz |
COMPMID-477 - Optimized batched case in CLConvolutionLayer
Change-Id: I4ef18f49f1da0cb816aaa0762466b940792c15ed
Reviewed-on: http://mpd-gerrit.cambridge.arm.com/84162
Tested-by: Kaizen <jeremy.johnson+kaizengerrit@arm.com>
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Diffstat (limited to 'tests/model_objects')
-rw-r--r-- | tests/model_objects/AlexNet.h | 4 |
1 files changed, 3 insertions, 1 deletions
diff --git a/tests/model_objects/AlexNet.h b/tests/model_objects/AlexNet.h index c9fd448d5d..45622e2118 100644 --- a/tests/model_objects/AlexNet.h +++ b/tests/model_objects/AlexNet.h @@ -24,6 +24,8 @@ #ifndef __ARM_COMPUTE_TEST_MODEL_OBJECTS_ALEXNET_H__ #define __ARM_COMPUTE_TEST_MODEL_OBJECTS_ALEXNET_H__ +#include "arm_compute/runtime/Tensor.h" + #include "tests/AssetsLibrary.h" #include "tests/Globals.h" #include "tests/Utils.h" @@ -149,7 +151,7 @@ public: b[6]->allocator()->init(TensorInfo(TensorShape(4096U), 1, dt, fixed_point_position)); b[7]->allocator()->init(TensorInfo(TensorShape(1000U), 1, dt, fixed_point_position)); - if(_batches > 1) + if(_batches > 1 && std::is_same<TensorType, Tensor>::value) { w[5]->allocator()->init(TensorInfo(TensorShape(9216U * dt_size, 4096U / dt_size), 1, dt, fixed_point_position)); w[6]->allocator()->init(TensorInfo(TensorShape(4096U * dt_size, 4096U / dt_size), 1, dt, fixed_point_position)); |