diff options
Diffstat (limited to 'tests/networks/AlexNetNetwork.h')
-rw-r--r-- | tests/networks/AlexNetNetwork.h | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tests/networks/AlexNetNetwork.h b/tests/networks/AlexNetNetwork.h index a30b7f8f75..97991b0121 100644 --- a/tests/networks/AlexNetNetwork.h +++ b/tests/networks/AlexNetNetwork.h @@ -107,7 +107,7 @@ public: { auto reshape = [&](unsigned int width, unsigned int height, bool convolution_layer) -> TensorShape { - const bool is_optimised = std::is_same<ITensorType, ITensor>::value && NEScheduler::get().cpu_info().CPU >= CPUTarget::ARMV7 && data_type == DataType::F32; + const bool is_optimised = std::is_same<ITensorType, ITensor>::value && data_type == DataType::F32; if(convolution_layer && is_optimised) { |