diff options
Diffstat (limited to 'src')
-rw-r--r-- | src/armnnCaffeParser/CaffeParser.cpp | 2 | ||||
-rw-r--r-- | src/armnnOnnxParser/OnnxParser.cpp | 2 | ||||
-rwxr-xr-x | src/armnnTfParser/TfParser.cpp | 2 |
3 files changed, 3 insertions, 3 deletions
diff --git a/src/armnnCaffeParser/CaffeParser.cpp b/src/armnnCaffeParser/CaffeParser.cpp index f95fe9c252..d11da466b8 100644 --- a/src/armnnCaffeParser/CaffeParser.cpp +++ b/src/armnnCaffeParser/CaffeParser.cpp @@ -1731,7 +1731,7 @@ INetworkPtr CaffeParser::CreateNetworkFromBinaryFile(const char* graphFile, FileInputStream inStream(fileno(fd)); CodedInputStream codedStream(&inStream); - codedStream.SetTotalBytesLimit(INT_MAX, INT_MAX); + codedStream.SetTotalBytesLimit(INT_MAX); bool success = netParam.ParseFromCodedStream(&codedStream); fclose(fd); diff --git a/src/armnnOnnxParser/OnnxParser.cpp b/src/armnnOnnxParser/OnnxParser.cpp index 4ae6627ac2..f3d0a73342 100644 --- a/src/armnnOnnxParser/OnnxParser.cpp +++ b/src/armnnOnnxParser/OnnxParser.cpp @@ -546,7 +546,7 @@ ModelPtr OnnxParser::LoadModelFromBinaryFile(const char* graphFile) google::protobuf::io::FileInputStream inStream(fileno(fd)); google::protobuf::io::CodedInputStream codedStream(&inStream); - codedStream.SetTotalBytesLimit(INT_MAX, INT_MAX); + codedStream.SetTotalBytesLimit(INT_MAX); bool success = modelProto.get()->ParseFromCodedStream(&codedStream); fclose(fd); diff --git a/src/armnnTfParser/TfParser.cpp b/src/armnnTfParser/TfParser.cpp index 255233bab3..aec8df8876 100755 --- a/src/armnnTfParser/TfParser.cpp +++ b/src/armnnTfParser/TfParser.cpp @@ -3564,7 +3564,7 @@ INetworkPtr TfParser::CreateNetworkFromBinaryFile(const char* graphFile, google::protobuf::io::FileInputStream inStream(fileno(fd)); google::protobuf::io::CodedInputStream codedStream(&inStream); - codedStream.SetTotalBytesLimit(INT_MAX, INT_MAX); + codedStream.SetTotalBytesLimit(INT_MAX); bool success = graphDef.ParseFromCodedStream(&codedStream); fclose(fd); |