[ Trivial ] add more error infor
authorjijoong.moon <jijoong.moon@samsung.com>
Mon, 20 Mar 2023 22:28:23 +0000 (07:28 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Tue, 21 Mar 2023 05:23:49 +0000 (14:23 +0900)
add more error infor when tflite and input dim mismatch.

**Self evaluation:**
1. Build test:  [X]Passed [ ]Failed [ ]Skipped
2. Run test:  [X]Passed [ ]Failed [ ]Skipped

Signed-off-by: jijoong.moon <jijoong.moon@samsung.com>
nntrainer/layers/tflite_layer.cpp

index 7189107..ca694df 100644 (file)
@@ -110,7 +110,8 @@ void TfLiteLayer::finalize(InitLayerContext &context) {
 
   for (size_t idx = 0; idx < dims.size(); idx++) {
     NNTR_THROW_IF(dims[idx] != input_dims[idx], std::invalid_argument)
-      << "Input dimensions mismatch";
+      << "Input dimensions mismatch -> " << idx << ":" << dims[idx] << " "
+      << input_dims[idx];
   }
 
   std::vector<TensorDim> output_dims;