onert::backend::builtin::kernel::IfLayer Class Reference

#include <IfLayer.h>

Collaboration diagram for onert::backend::builtin::kernel::IfLayer:

Public Member Functions
	IfLayer (backend::IPortableTensor *cond_tensor, const std::vector< backend::IPortableTensor * > input_tensors, const std::vector< backend::IPortableTensor * > output_tensors, const ir::SubgraphIndex &then_subg_index, const ir::SubgraphIndex &else_subg_index, exec::IExecutors *executors, const ir::ModelIndex &model_index, const std::shared_ptr< ExternalContext > &external_context)
void	run () override
Public Member Functions inherited from onert::exec::IFunction
virtual	~IFunction ()=default
virtual void	prepare ()

Detailed Description

Definition at line 28 of file IfLayer.h.

Constructor & Destructor Documentation

IfLayer()

onert::backend::builtin::kernel::IfLayer::IfLayer	(	backend::IPortableTensor *	cond_tensor,
		const std::vector< backend::IPortableTensor * >	input_tensors,
		const std::vector< backend::IPortableTensor * >	output_tensors,
		const ir::SubgraphIndex &	then_subg_index,
		const ir::SubgraphIndex &	else_subg_index,
		exec::IExecutors *	executors,
		const ir::ModelIndex &	model_index,
		const std::shared_ptr< ExternalContext > &	external_context
	)

Definition at line 22 of file IfLayer.cc.

  : _cond_tensor{cond_tensor}, _input_tensors{input_tensors}, _output_tensors{output_tensors},
    _then_subg_index{then_subg_index}, _else_subg_index{else_subg_index}, _executors{executors},
    _model_index{model_index}, _external_context{external_context}
{
  // At this point, executors may not have executors of then subg and else subg
}

Member Function Documentation

run()

void onert::backend::builtin::kernel::IfLayer::run ( )

overridevirtual

Implements onert::exec::IFunction.

Definition at line 35 of file IfLayer.cc.

{
  // Check condition
  // // If true
  // // // Set _input_tensors -> then-subg's inputs
  // // // Set outputs of then-subg -> _output_tensors
  // // // Run then-subg
  // // Else
  // // // Set _input_tensors -> else-subg's inputs
  // // // Set outputs of else-subg -> _output_tensors
  // // // Run else-subg
 
  auto getResultCond = [](backend::IPortableTensor *tensor) -> bool {
    bool ret = false;
    tensor->access([&](ITensor &tensor) { ret = *reinterpret_cast<bool *>(tensor.buffer()); });
    return ret;
  };
 
  exec::IExecutor *subg_exec = nullptr;
  bool cond_result = getResultCond(_cond_tensor);
  if (cond_result)
  {
    VERBOSE(If) << "Call to $" << _then_subg_index << " (then)" << std::endl;
    subg_exec = _executors->at(_model_index, _then_subg_index);
  }
  else
  {
    VERBOSE(If) << "Call to $" << _else_subg_index << " (else)" << std::endl;
    subg_exec = _executors->at(_model_index, _else_subg_index);
  }
 
  subg_exec->execute(_input_tensors, _output_tensors,
                     _executors->entryExecutor()->currentOptions());
  VERBOSE(If) << "Return from $" << (cond_result ? _then_subg_index : _else_subg_index)
              << std::endl;
}

References onert::exec::IExecutors::at(), onert::exec::IExecutor::currentOptions(), onert::exec::IExecutors::entryExecutor(), onert::exec::IExecutor::execute(), and VERBOSE.

---

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

• runtime/onert/core/src/backend/builtin/kernel/IfLayer.h
• runtime/onert/core/src/backend/builtin/kernel/IfLayer.cc