#include <While.h>

Collaboration diagram for luci_interpreter::kernels::While:

Public Member Functions
	While (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs, RuntimeGraph cond_graph, RuntimeGraph body_graph)

const Tensor *	input (int index) const

Tensor *	output (int index) const

void	configure () override

void	execute () const override

Public Member Functions inherited from luci_interpreter::Kernel
virtual	~Kernel ()=default

const std::vector< const Tensor * > &	getInputTensors () const

const std::vector< Tensor * > &	getOutputTensors () const

Additional Inherited Members
Protected Member Functions inherited from luci_interpreter::Kernel
	Kernel (std::vector< const Tensor * > inputs, std::vector< Tensor * > outputs)

Protected Attributes inherited from luci_interpreter::Kernel
const std::vector< const Tensor * >	_inputs

const std::vector< Tensor * >	_outputs

Detailed Description

Definition at line 28 of file While.h.

Constructor & Destructor Documentation

◆ While()

luci_interpreter::kernels::While::While	(	std::vector< const Tensor * >	inputs,
		std::vector< Tensor * >	outputs,
		RuntimeGraph *	cond_graph,
		RuntimeGraph *	body_graph
	)

Definition at line 61 of file While.cpp.

  : Kernel(std::move(inputs), std::move(outputs)), _cond_graph(cond_graph), _body_graph(body_graph)
{
}

Member Function Documentation

◆ configure()

void luci_interpreter::kernels::While::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 67 of file While.cpp.

{
  LUCI_INTERPRETER_CHECK(_body_graph->getInputTensors().size() == getInputTensors().size());
  LUCI_INTERPRETER_CHECK(_body_graph->getOutputTensors().size() == getOutputTensors().size());
  LUCI_INTERPRETER_CHECK(_body_graph->getOutputTensors().size() == getInputTensors().size());
 
  LUCI_INTERPRETER_CHECK(_cond_graph->getInputTensors().size() == getInputTensors().size());
 
  const auto &cond_outputs = _cond_graph->getOutputTensors();
  LUCI_INTERPRETER_CHECK(cond_outputs.size() == 1)
  LUCI_INTERPRETER_CHECK(cond_outputs[0]->element_type() == DataType::BOOL);
}

References luci_interpreter::Kernel::getInputTensors(), luci_interpreter::RuntimeGraph::getInputTensors(), luci_interpreter::Kernel::getOutputTensors(), luci_interpreter::RuntimeGraph::getOutputTensors(), LUCI_INTERPRETER_CHECK, and size.

◆ execute()

void luci_interpreter::kernels::While::execute ( ) const

overridevirtual

Note: Dynamic shape such as {1, 0, 8} may fail in tensor->data()

Implements luci_interpreter::Kernel.

Definition at line 83 of file While.cpp.

{
  const auto &cond_inputs = _cond_graph->getInputTensors();
  const auto &cond_outputs = _cond_graph->getOutputTensors();
 
  configureTensorsAllocations(cond_inputs, _cond_graph);
 
  copy(getInputTensors(), cond_inputs);
 
  const auto &body_inputs = _body_graph->getInputTensors();
  const auto &body_outputs = _body_graph->getOutputTensors();
 
  configureTensorsAllocations(body_inputs, _body_graph);
 
  while (true)
  {
    _cond_graph->execute();
 
    bool cond_value = cond_outputs[0]->data<bool>()[0];
    if (!cond_value)
      break;
 
    copy(cond_inputs, body_inputs);
 
    _body_graph->execute();
 
    copy(body_outputs, cond_inputs);
  }
 
  copy(cond_inputs, getOutputTensors());
}