diff options
Diffstat (limited to 'inference-engine/include/cpp/ie_cnn_net_reader.h')
-rw-r--r-- | inference-engine/include/cpp/ie_cnn_net_reader.h | 6 |
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(); } |