#include <Model.h>

Public Member Functions
	Model ()=default

	Model (const Model &obj)=default

	Model (Model &&)=default

Model &	operator= (const Model &)=default

Model &	operator= (Model &&)=default

	~Model ()=default

void	push (SubgraphIndex index, const std::shared_ptr< IGraph > &subg)
	Put subgraph in the container with a new Index for that.

void	remove (const SubgraphIndex &index)
	Remove the subgraph that is associated with the given index.

const std::shared_ptr< IGraph > &	at (const SubgraphIndex &index) const
	Get the subgraph that is associated with the given index.

std::shared_ptr< IGraph > &	at (const SubgraphIndex &index)
	Get the subgraph that is associated with the given index.

bool	exist (const SubgraphIndex &index) const
	Get the subgraph that is associated with the given index.

void	iterate (const std::function< void(const SubgraphIndex &, const IGraph &)> &fn) const
	Iterate over the container with given function.

void	iterate (const std::function< void(const SubgraphIndex &, IGraph &)> &fn)
	Iterate over the container with given function.

size_t	subgraphs_count () const
	Get count of Subgraphs.

std::shared_ptr< IGraph >	primary_subgraph () const
	Return the primary subgraph.

template<typename Graph , std::enable_if_t< std::is_base_of_v< IGraph, Graph >, bool > = true>
bool	hasOnly ()
	Return whether the model has only typename Graph.

void	bindKernelBuilder (const std::shared_ptr< onert::backend::custom::IKernelBuilder > &kernel_builder)

const std::shared_ptr< backend::custom::IKernelBuilder > &	getKernelBuilder () const

void	add_metadata (const std::string &name, std::unique_ptr< const ir::Data > data)

bool	exists_metadata (const std::string &name) const

std::unique_ptr< const ir::Data >	extract_metadata (const std::string name)

Detailed Description

Definition at line 35 of file Model.h.

Constructor & Destructor Documentation

◆ Model() [1/3]

onert::ir::Model::Model ( )

default

◆ Model() [2/3]

onert::ir::Model::Model ( const Model & obj )

default

◆ Model() [3/3]

onert::ir::Model::Model ( Model && )

default

◆ ~Model()

onert::ir::Model::~Model ( )

default

Member Function Documentation

◆ add_metadata()

void onert::ir::Model::add_metadata	(	const std::string &	name,
		std::unique_ptr< const ir::Data >	data
	)

inline

Definition at line 172 of file Model.h.

  {
    _metadatas.emplace(name, std::move(data));
  }

◆ at() [1/2]

std::shared_ptr< IGraph > & onert::ir::Model::at ( const SubgraphIndex & index )

inline

Get the subgraph that is associated with the given index.

Parameters

[in] index Index of the subgraph to be returned

Returns: IGraph

Definition at line 78 of file Model.h.

78{ return _subgraphs.at(index); }

◆ at() [2/2]

const std::shared_ptr< IGraph > & onert::ir::Model::at ( const SubgraphIndex & index ) const

inline

Get the subgraph that is associated with the given index.

Parameters

[in] index Index of the subgraph to be returned

Returns: IGraph

Definition at line 68 of file Model.h.

  {
    return _subgraphs.at(index);
  }

◆ bindKernelBuilder()

void onert::ir::Model::bindKernelBuilder ( const std::shared_ptr< onert::backend::custom::IKernelBuilder > & kernel_builder )

inline

Definition at line 158 of file Model.h.

  {
    _kernel_builder = kernel_builder;
  }

◆ exist()

bool onert::ir::Model::exist ( const SubgraphIndex & index ) const

inline

Get the subgraph that is associated with the given index.

Parameters

[in] index Index of the subgraph to be returned

Returns: true if such entry exists otherwise false

Definition at line 86 of file Model.h.

  {
    auto it = _subgraphs.find(index);
    return it != _subgraphs.end();
  }

◆ exists_metadata()

bool onert::ir::Model::exists_metadata ( const std::string & name ) const

inline

Definition at line 177 of file Model.h.

  {
    return _metadatas.find(name) != _metadatas.end();
  }

◆ extract_metadata()

std::unique_ptr< const ir::Data > onert::ir::Model::extract_metadata ( const std::string name )

inline

Definition at line 183 of file Model.h.

  {
    auto m = _metadatas.find(name);
 
    if (m == _metadatas.end())
      throw std::out_of_range{"no meatdata named " + name};
 
    auto data = std::move(m->second);
    _metadatas.erase(m);
    return data;
  }

References m.

◆ getKernelBuilder()

const std::shared_ptr< backend::custom::IKernelBuilder > & onert::ir::Model::getKernelBuilder ( ) const

inline

Definition at line 163 of file Model.h.

  {
    return _kernel_builder;
  }

◆ hasOnly()

template<typename Graph , std::enable_if_t< std::is_base_of_v< IGraph, Graph >, bool > = true>

bool onert::ir::Model::hasOnly ( )

inline

Return whether the model has only typename Graph.

Template Parameters

Graph Type that inherits from IGraph

Returns: true if the model has only typename Graph, otherwise false

Definition at line 142 of file Model.h.

  {
    for (const auto &e : _subgraphs)
    {
      if (std::dynamic_pointer_cast<Graph>(e.second) == nullptr)
        return false;
    }
    return true;
  }

◆ iterate() [1/2]

void onert::ir::Model::iterate ( const std::function< void(const SubgraphIndex &, const IGraph &)> & fn ) const

inline

Iterate over the container with given function.

Parameters

[in] fn Function to be run for every container entry

Returns: N/A

Definition at line 98 of file Model.h.

  {
    for (const auto &[subg_idx, subg] : _subgraphs)
    {
      fn(subg_idx, *subg);
    }
  }

◆ iterate() [2/2]

void onert::ir::Model::iterate ( const std::function< void(const SubgraphIndex &, IGraph &)> & fn )

inline

Iterate over the container with given function.

Parameters

[in] fn Function to be run for every container entry

Returns: N/A

Definition at line 112 of file Model.h.

  {
    for (const auto &[subg_idx, subg] : _subgraphs)
    {
      fn(subg_idx, *subg);
    }
  }

◆ operator=() [1/2]

Model & onert::ir::Model::operator= ( const Model & )

default

◆ operator=() [2/2]

Model & onert::ir::Model::operator= ( Model && )

default

◆ primary_subgraph()

std::shared_ptr< IGraph > onert::ir::Model::primary_subgraph ( ) const

inline

Return the primary subgraph.

Returns: std::shared_ptr<IGraph> Primary subgraph

Definition at line 132 of file Model.h.

132{ return _subgraphs.at(SubgraphIndex{0}); }

onert::ir::SubgraphIndex

::onert::util::Index< uint16_t, SubgraphIndexTag > SubgraphIndex

Definition Index.h:39

◆ push()

void onert::ir::Model::push	(	SubgraphIndex	index,
		const std::shared_ptr< IGraph > &	subg
	)

inline

Put subgraph in the container with a new Index for that.

Parameters

[in]	subg	Subgraph to be pushed
[in]	index	Index of subgraph to be pushed

Returns: Created

Definition at line 52 of file Model.h.

52{ _subgraphs[index] = subg; }

moco::index

loco::GraphInputIndex index(const TFPlaceholder *node)

Definition TFNode.cpp:54

◆ remove()

void onert::ir::Model::remove ( const SubgraphIndex & index )

inline

Remove the subgraph that is associated with the given index.

Parameters

[in] index Index of the subgraph to be removed

Returns: N/A

Definition at line 60 of file Model.h.

60{ _subgraphs.erase(index); }

◆ subgraphs_count()

size_t onert::ir::Model::subgraphs_count ( ) const

inline

Get count of Subgraphs.

Returns: count of Subgraphs

Definition at line 125 of file Model.h.

125{ return _subgraphs.size(); }

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

runtime/onert/core/include/ir/Model.h

Public Member Functions

Detailed Description

Constructor & Destructor Documentation

◆ Model() [1/3]

◆ Model() [2/3]

◆ Model() [3/3]

◆ ~Model()

Member Function Documentation

◆ add_metadata()

◆ at() [1/2]

◆ at() [2/2]

◆ bindKernelBuilder()

◆ exist()

◆ exists_metadata()

◆ extract_metadata()

◆ getKernelBuilder()

◆ hasOnly()

◆ iterate() [1/2]

◆ iterate() [2/2]

◆ operator=() [1/2]

◆ operator=() [2/2]

◆ primary_subgraph()

◆ push()

◆ remove()

◆ subgraphs_count()