ExecuteNetwork.cpp File Reference

#include "ExecuteNetworkProgramOptions.hpp"
#include "ArmNNExecutor.hpp"
#include <armnn/Logging.hpp>

Go to the source code of this file.

Functions
std::unique_ptr< IExecutor >	BuildExecutor (ProgramOptions &programOptions)
int	main (int argc, const char *argv[])

Function Documentation

BuildExecutor()

std::unique_ptr<IExecutor> BuildExecutor

(

ProgramOptions &

programOptions

)

Definition at line 14 of file ExecuteNetwork.cpp.

References ARMNN_LOG, ExecuteNetworkParams::ArmNNTfLiteDelegate, ProgramOptions::m_ExNetParams, ProgramOptions::m_RuntimeOptions, ExecuteNetworkParams::m_TfLiteExecutor, and ExecuteNetworkParams::TfliteInterpreter.

Referenced by main().

{
    if (programOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::ArmNNTfLiteDelegate ||
        programOptions.m_ExNetParams.m_TfLiteExecutor == ExecuteNetworkParams::TfLiteExecutor::TfliteInterpreter)
    {
#if defined(ARMNN_TFLITE_DELEGATE)
        return std::make_unique<TfLiteExecutor>(programOptions.m_ExNetParams);
#else
        ARMNN_LOG(fatal) << "Not built with Arm NN Tensorflow-Lite delegate support.";
            return nullptr;
#endif
    }
    else
    {
        return std::make_unique<ArmNNExecutor>(programOptions.m_ExNetParams, programOptions.m_RuntimeOptions);
    }
}

main()

int main	(	int	argc,
		const char *	argv[]
	)

Definition at line 33 of file ExecuteNetwork.cpp.

References ARMNN_LOG, BuildExecutor(), armnn::ConfigureLogging(), armnn::Debug, armnn::Info, ExecuteNetworkParams::m_CompareWithTflite, ExecuteNetworkParams::m_ComparisonComputeDevices, ExecuteNetworkParams::m_ComputeDevices, ProgramOptions::m_ExNetParams, ExecuteNetworkParams::m_TfLiteExecutor, ProgramOptions::ParseOptions(), and ExecuteNetworkParams::TfliteInterpreter.

{
    // Configures logging for both the ARMNN library and this test program.
#ifdef NDEBUG
    armnn::LogSeverity level = armnn::LogSeverity::Info;
#else
    armnn::LogSeverity level = armnn::LogSeverity::Debug;
#endif
    armnn::ConfigureLogging(true, true, level);


    // Get ExecuteNetwork parameters and runtime options from command line
    // This might throw an InvalidArgumentException if the user provided invalid inputs
    ProgramOptions programOptions;
    try
    {
        programOptions.ParseOptions(argc, argv);
    }
    catch (const std::exception& e)
    {
        ARMNN_LOG(fatal) << e.what();
        return EXIT_FAILURE;
    }

    std::vector<const void*> outputResults;

    auto executor = BuildExecutor(programOptions);
    if (!executor)
    {
        return EXIT_FAILURE;
    }

    executor->PrintNetworkInfo();
    outputResults = executor->Execute();

    if (!programOptions.m_ExNetParams.m_ComparisonComputeDevices.empty() ||
         programOptions.m_ExNetParams.m_CompareWithTflite)
    {
        ExecuteNetworkParams comparisonParams = programOptions.m_ExNetParams;
        comparisonParams.m_ComputeDevices = programOptions.m_ExNetParams.m_ComparisonComputeDevices;

        if (programOptions.m_ExNetParams.m_CompareWithTflite)
        {
            comparisonParams.m_TfLiteExecutor = ExecuteNetworkParams::TfLiteExecutor::TfliteInterpreter;
        }

        auto comparisonExecutor = BuildExecutor(programOptions);

        if (!comparisonExecutor)
        {
            return EXIT_FAILURE;
        }

        comparisonExecutor->PrintNetworkInfo();
        comparisonExecutor->Execute();

        comparisonExecutor->CompareAndPrintResult(outputResults);
    }
}