commit 415a2bfef929439783dfdba03fd67e9701c4822c
author Georgios Pinitas <georgios.pinitas@arm.com> Mon Jul 30 12:05:25 2018 +0100
committer Anthony Barbier <anthony.barbier@arm.com> Fri Nov 02 16:54:54 2018 +0000
tree 1068a197ea9830e0d154ff3a348644f21a4af1e6
parent 68a3f56627b04acdefebe67d645727dd83889766

COMPMID-1432: Enable by default NHWC in graph examples

Change-Id: I2240b6a6430cb1d261458343b2900cc1f16ac414
Reviewed-on: https://eu-gerrit-1.euhpc.arm.com/141861
Reviewed-by: Anthony Barbier <anthony.barbier@arm.com>
Reviewed-by: Gian Marco Iodice <gianmarco.iodice@arm.com>
Tested-by: Jenkins <bsgcomp@arm.com>

examples/graph_alexnet.cpp

diff --git a/examples/graph_alexnet.cpp b/examples/graph_alexnet.cpp
index 944a435..8394587 100644
--- a/examples/graph_alexnet.cpp
+++ b/examples/graph_alexnet.cpp
@@ -58,6 +58,12 @@
             return false;
         }
 
+        // Set default layout if needed
+        if(!common_opts.data_layout->is_set() && common_params.target == Target::NEON)
+        {
+            common_params.data_layout = DataLayout::NCHW;
+        }
+
         // Checks
         ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "Unsupported data type!");