From 6dd178f2395b34cfb360eabb0130c19ed258f5fa Mon Sep 17 00:00:00 2001 From: Nikhil Raj Date: Fri, 2 Apr 2021 22:04:39 +0100 Subject: IVGCVSW-5720 Remove the Caffe Parser from ArmNN Signed-off-by: Nikhil Raj Change-Id: Ib00be204f549efa9aa5971ecf65c2dec4a10b10f --- tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) (limited to 'tests/MultipleNetworksCifar10/MultipleNetworksCifar10.cpp') 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 networks; - armnnCaffeParser::ICaffeParserPtr parser(armnnCaffeParser::ICaffeParser::Create()); + armnnTfParser::ITfParserPtr parser(armnnTfParser::ITfParser::Create()); const int networksCount = 4; for (int i = 0; i < networksCount; ++i) -- cgit v1.2.1