feat: error message display when error happens when running the graph.

MRKonrad · Apr 18, 2019 · 74158ae · 74158ae
1 parent 3d9c087
commit 74158ae
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/oxtfUtils.cpp b/src/oxtfUtils.cpp
@@ -194,7 +194,8 @@ int sessionRun(TF_Tensor* input_tensor, TF_Tensor*& output_tensor, std::string g
     );
 
     if (TF_GetCode(status) != TF_OK) {
-        std::cout << "Error " << TF_GetCode(status) << " running session";
+        std::cout << "Error running session: TF_Code: " << TF_GetCode(status)
+                  << " message: " << TF_Message(status) << std::endl;
         TF_DeleteStatus(status);
         return 5;
     }