Entry.cpp File Reference

#include "ConvBackend.h"
#include <nnsuite/conv/RandomModel.h>
#include <nnkit/Backend.h>
#include <nnkit/CmdlineArguments.h>
#include <memory>
#include <chrono>
#include <iostream>

Go to the source code of this file.

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

Function Documentation

make_backend()

std::unique_ptr< nnkit::Backend > make_backend

(

const nnkit::CmdlineArguments &

args

)

Definition at line 28 of file Entry.cpp.

{
  // Set random seed
  int32_t seed = std::chrono::system_clock::now().time_since_epoch().count();
 
  if (args.size() > 0)
  {
    seed = std::stoi(args.at(0), nullptr, 0);
  }
 
  std::cout << "SEED: " << seed << std::endl;
 
  const nnsuite::conv::RandomModel model{seed};
 
  return std::make_unique<ConvBackend>(model);
}