onert::compiler::CompilerFactory Class Reference

#include <CompilerFactory.h>

Public Member Functions
std::unique_ptr< ICompiler >	create (const std::shared_ptr< ir::NNPkg > &nnpkg, CompilerOptions *copts, const ir::train::TrainingInfo *training_info=nullptr)

Static Public Member Functions
static CompilerFactory &	get ()

Detailed Description

Definition at line 29 of file CompilerFactory.h.

Member Function Documentation

create()

std::unique_ptr< ICompiler > onert::compiler::CompilerFactory::create	(	const std::shared_ptr< ir::NNPkg > &	nnpkg,
		CompilerOptions *	copts,
		const ir::train::TrainingInfo *	training_info = nullptr
	)

Definition at line 32 of file CompilerFactory.cc.

{
  // Returing compiler for training
  if (training_info)
    return std::make_unique<train::TrainingCompiler>(nnpkg, copts, *training_info);
 
  // Returing compiler for inference
  if (nnpkg->model_count() == 1)
    return std::make_unique<Compiler>(nnpkg, copts);
 
  return std::make_unique<MultiModelCompiler>(nnpkg, copts);
}

Referenced by nnfw_session::prepare().

get()

CompilerFactory & onert::compiler::CompilerFactory::get ( )

static

Definition at line 26 of file CompilerFactory.cc.

{
  static CompilerFactory singleton;
  return singleton;
}

Referenced by nnfw_session::prepare().

---

The documentation for this class was generated from the following files:

• runtime/onert/core/include/compiler/CompilerFactory.h
• runtime/onert/core/src/compiler/CompilerFactory.cc