summaryrefslogtreecommitdiff
path: root/inference-engine/include/cpp/ie_cnn_net_reader.h
diff options
context:
space:
mode:
Diffstat (limited to 'inference-engine/include/cpp/ie_cnn_net_reader.h')
-rw-r--r--inference-engine/include/cpp/ie_cnn_net_reader.h6
1 files changed, 5 insertions, 1 deletions
diff --git a/inference-engine/include/cpp/ie_cnn_net_reader.h b/inference-engine/include/cpp/ie_cnn_net_reader.h
index 6a15c810b..5a9b55a1d 100644
--- a/inference-engine/include/cpp/ie_cnn_net_reader.h
+++ b/inference-engine/include/cpp/ie_cnn_net_reader.h
@@ -75,7 +75,11 @@ public:
CNNNetwork getNetwork() {
// network obj are to be updated upon this call
if (network.get() == nullptr) {
- network.reset(new CNNNetwork(actual));
+ try {
+ network.reset(new CNNNetwork(actual));
+ } catch (...) {
+ THROW_IE_EXCEPTION << "Could not allocate memory";
+ }
}
return *network.get();
}