diff options
Diffstat (limited to 'src/armnnCaffeParser')
-rw-r--r-- | src/armnnCaffeParser/RecordByRecordCaffeParser.cpp | 9 |
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); } - - - |