diff options
Diffstat (limited to 'examples/graph_vgg19.cpp')
| -rw-r--r-- | examples/graph_vgg19.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/graph_vgg19.cpp b/examples/graph_vgg19.cpp index 8bf88b96ed..8de0223a48 100644 --- a/examples/graph_vgg19.cpp +++ b/examples/graph_vgg19.cpp @@ -58,7 +58,8 @@ public: } // Checks - ARM_COMPUTE_ERROR_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "Unsupported data type!"); + ARM_COMPUTE_EXIT_ON_MSG(arm_compute::is_data_type_quantized_asymmetric(common_params.data_type), "Unsupported data type!"); + ARM_COMPUTE_EXIT_ON_MSG(common_params.data_layout == DataLayout::NHWC, "Unsupported data layout!"); // Print parameter values std::cout << common_params << std::endl; |