aboutsummaryrefslogtreecommitdiff
path: root/src/armnnCaffeParser
diff options
context:
space:
mode:
Diffstat (limited to 'src/armnnCaffeParser')
-rw-r--r--src/armnnCaffeParser/RecordByRecordCaffeParser.cpp9
1 files changed, 5 insertions, 4 deletions
diff --git a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
index 592dc5c0de..cb7943655d 100644
--- a/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
+++ b/src/armnnCaffeParser/RecordByRecordCaffeParser.cpp
@@ -474,8 +474,12 @@ armnn::INetworkPtr RecordByRecordCaffeParser::CreateNetworkFromBinaryFile(
}
m_RequestedOutputs = requestedOutputs;
- //FILE * fp = fopen(graphFile, "rb");
std::ifstream ifs(graphFile, std::ifstream::in|std::ifstream::binary);
+ if (ifs.fail())
+ {
+ throw armnn::FileNotFoundException("Failed to open graph file '" + std::string(graphFile) + "'");
+ }
+
std::vector<LayerParameterInfo> layerInfo;
NetParameterInfo netParameterInfo;
while(true)
@@ -727,6 +731,3 @@ armnn::INetworkPtr RecordByRecordCaffeParser::LoadLayers(std::ifstream& ifs,
return move(m_Network);
}
-
-
-