luci_interpreter::kernels::If Class Reference

#include <If.h>

Collaboration diagram for luci_interpreter::kernels::If:

Public Member Functions
	If (const Tensor *cond, const std::vector< const Tensor * > &inputs, std::vector< Tensor * > outputs, RuntimeGraph *then_graph, RuntimeGraph *else_graph)
const Tensor *	cond () const
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 If.h.

Constructor & Destructor Documentation

If()

luci_interpreter::kernels::If::If	(	const Tensor *	cond,
		const std::vector< const Tensor * > &	inputs,
		std::vector< Tensor * >	outputs,
		RuntimeGraph *	then_graph,
		RuntimeGraph *	else_graph
	)

Definition at line 35 of file If.cpp.

  : Kernel(joinInputs(cond, inputs), std::move(outputs)), _then_graph(then_graph),
    _else_graph(else_graph)
{
}

Member Function Documentation

cond()

const Tensor * luci_interpreter::kernels::If::cond ( ) const

inline

Definition at line 34 of file If.h.

{ return _inputs[0]; }

References luci_interpreter::Kernel::_inputs.

Referenced by configure(), and execute().

configure()

void luci_interpreter::kernels::If::configure ( )

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 42 of file If.cpp.

{
  LUCI_INTERPRETER_CHECK(cond()->element_type() == DataType::BOOL);
  LUCI_INTERPRETER_CHECK(cond()->shape().num_elements() == 1);
 
  for (RuntimeGraph *graph : {_then_graph, _else_graph})
  {
    (void)graph;
    LUCI_INTERPRETER_CHECK(graph->getInputTensors().size() == getInputTensors().size() - 1);
    LUCI_INTERPRETER_CHECK(graph->getOutputTensors().size() == getOutputTensors().size());
  }
}

References cond(), luci_interpreter::Kernel::getInputTensors(), luci_interpreter::Kernel::getOutputTensors(), LUCI_INTERPRETER_CHECK, luci::must_cast(), and size.

execute()

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

overridevirtual

Implements luci_interpreter::Kernel.

Definition at line 55 of file If.cpp.

{
  const bool cond_value = cond()->data<bool>()[0];
 
  RuntimeGraph *active_graph = cond_value ? _then_graph : _else_graph;
  const auto &graph_inputs = active_graph->getInputTensors();
  const auto &graph_outputs = active_graph->getOutputTensors();
 
  // Copy kernel inputs to active graph inputs.
  for (size_t i = 0; i < getInputTensors().size() - 1; ++i)
  {
    LUCI_INTERPRETER_CHECK(graph_inputs[i]->element_type() == input(i)->element_type());
    graph_inputs[i]->resize(input(i)->shape());
 
    const int32_t num_elements = input(i)->shape().num_elements();
    const std::size_t element_size = getDataTypeSize(input(i)->element_type());
    // TODO: Think about how allocate memory for output in main graph
    active_graph->configureAllocations(graph_inputs[i]);
    std::memcpy(graph_inputs[i]->data<void>(), input(i)->data<void>(), num_elements * element_size);
  }
 
  active_graph->execute();
 
  // Copy graph outputs to kernel outputs.
  for (size_t i = 0; i < getOutputTensors().size(); ++i)
  {
    LUCI_INTERPRETER_CHECK(graph_outputs[i]->element_type() == output(i)->element_type());
    output(i)->resize(graph_outputs[i]->shape());
    // TODO: Think about how allocate memory for output in main graph
    active_graph->configureAllocations(output(i));
 
    const int32_t num_elements = output(i)->shape().num_elements();
    const std::size_t element_size = getDataTypeSize(output(i)->element_type());
    std::memcpy(output(i)->data<void>(), graph_outputs[i]->data<void>(),
                num_elements * element_size);
  }
}

References cond(), luci_interpreter::Tensor::data(), luci_interpreter::getDataTypeSize(), luci_interpreter::Kernel::getInputTensors(), luci_interpreter::RuntimeGraph::getInputTensors(), luci_interpreter::Kernel::getOutputTensors(), input(), LUCI_INTERPRETER_CHECK, luci::must_cast(), luci_interpreter::Shape::num_elements(), output(), luci_interpreter::Tensor::resize(), and luci_interpreter::Tensor::shape().

input()

const Tensor * luci_interpreter::kernels::If::input ( int  index ) const

inline

Definition at line 35 of file If.h.

{ return _inputs[1 + index]; }

References luci_interpreter::Kernel::_inputs.

Referenced by execute().

output()

Tensor * luci_interpreter::kernels::If::output ( int  index ) const

inline

Definition at line 36 of file If.h.

{ return _outputs[index]; }

References luci_interpreter::Kernel::_outputs.

Referenced by execute().

---

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

• compiler/luci-interpreter/src/kernels/If.h
• compiler/luci-interpreter/src/kernels/If.cpp