ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Backend.cpp File Reference
#include "nnkit/support/tf/Backend.h"
#include <nnkit/CmdlineArguments.h>
#include <memory>
#include <cassert>

Go to the source code of this file.

Functions

std::unique_ptr< nnkit::Backendmake_backend (const nnkit::CmdlineArguments &args)
 

Function Documentation

◆ make_backend()

std::unique_ptr< nnkit::Backend > make_backend ( const nnkit::CmdlineArguments args)

Definition at line 24 of file Backend.cpp.

25{
26 using std::make_unique;
27
28 assert(args.size() == 2); // args.at[0] : test.pb path, argas.at[1]: test.info path
29
30 return make_unique<::nnkit::support::tf::Backend>(args.at(0), args.at(1));
31}
args
Definition infer.py:21