diff options
author | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-02 22:04:39 +0100 |
---|---|---|
committer | Nikhil Raj <nikhil.raj@arm.com> | 2021-04-16 14:27:27 +0100 |
commit | 6dd178f2395b34cfb360eabb0130c19ed258f5fa (patch) | |
tree | 6814e4729bbd1e652d8b9c18c9db28f3bc2f8a8a /tests/MultipleNetworksCifar10 | |
parent | ed7fce413410d15c501ea52f9e6bfbbf71b3daf1 (diff) | |
download | armnn-6dd178f2395b34cfb360eabb0130c19ed258f5fa.tar.gz |
IVGCVSW-5720 Remove the Caffe Parser from ArmNN
Signed-off-by: Nikhil Raj <nikhil.raj@arm.com>
Change-Id: Ib00be204f549efa9aa5971ecf65c2dec4a10b10f
Diffstat (limited to 'tests/MultipleNetworksCifar10')
-rw-r--r-- | tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp index 456ff68e7c..9c51d3f0a7 100644 --- a/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp +++ b/tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp @@ -6,7 +6,7 @@ #include "armnn/ArmNN.hpp" #include "armnn/Utils.hpp" #include "armnn/INetwork.hpp" -#include "armnnCaffeParser/ICaffeParser.hpp" +#include "armnnTfParser/TfParser.hpp" #include "../Cifar10Database.hpp" #include "../InferenceTest.hpp" #include "../InferenceModel.hpp" @@ -89,7 +89,7 @@ int main(int argc, char* argv[]) return EXIT_FAILURE; } - fs::path modelPath = fs::path(modelDir + "/cifar10_full_iter_60000.caffemodel"); + fs::path modelPath = fs::path(modelDir + "/cifar10_tf.prototxt"); // Create runtime // This will also load dynamic backend in case that the dynamic backend path is specified @@ -123,7 +123,7 @@ int main(int argc, char* argv[]) }; std::vector<Net> networks; - armnnCaffeParser::ICaffeParserPtr parser(armnnCaffeParser::ICaffeParser::Create()); + armnnTfParser::ITfParserPtr parser(armnnTfParser::ITfParser::Create()); const int networksCount = 4; for (int i = 0; i < networksCount; ++i) |