diff options
Diffstat (limited to 'examples/graph_resnet50.cpp')
-rw-r--r-- | examples/graph_resnet50.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/examples/graph_resnet50.cpp b/examples/graph_resnet50.cpp index 66fc6e869d..2970037768 100644 --- a/examples/graph_resnet50.cpp +++ b/examples/graph_resnet50.cpp @@ -59,7 +59,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; |