#include <KernelBuilder.h>

Collaboration diagram for luci_interpreter::KernelBuilder:

Public Member Functions
	KernelBuilder (const std::unordered_map< const loco::Graph , RuntimeGraph > &graph_to_runtime_graph, const std::unordered_map< const loco::Node , Tensor > &node_to_tensor)

	~KernelBuilder ()

std::unique_ptr< Kernel >	build (const luci::CircleNode *node)

Public Member Functions inherited from luci_interpreter::KernelBuilderHelper
	KernelBuilderHelper (const std::unordered_map< const loco::Graph , RuntimeGraph > &graph_to_runtime_graph, const std::unordered_map< const loco::Node , Tensor > &node_to_tensor)

const Tensor *	getInputTensor (const loco::Node *node) const

const Tensor *	getOptionalInputTensor (const loco::Node *node) const

Tensor *	getOutputTensor (const loco::Node *node) const

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

RuntimeGraph *	getRuntimeGraph (const loco::Graph *graph) const

const std::unordered_map< const loco::Graph , RuntimeGraph > &	graph_to_runtime_graph () const

const std::unordered_map< const loco::Node , Tensor > &	node_to_tensor () const

Detailed Description

Definition at line 35 of file KernelBuilder.h.

Constructor & Destructor Documentation

◆ KernelBuilder()

luci_interpreter::KernelBuilder::KernelBuilder	(	const std::unordered_map< const loco::Graph , RuntimeGraph > &	graph_to_runtime_graph,
		const std::unordered_map< const loco::Node , Tensor > &	node_to_tensor
	)

Definition at line 107 of file KernelBuilder.cpp.

  : KernelBuilderHelper(graph_to_runtime_graph, node_to_tensor)
{
  _builder_registry = std::make_unique<KernelBuilderRegistry>();
}

◆ ~KernelBuilder()

luci_interpreter::KernelBuilder::~KernelBuilder ( )

Definition at line 115 of file KernelBuilder.cpp.

{
  // Need to define in this CPP to hide KernelBuilderRegistry internals.
  // This destructor deletes _builder_registry
}

Member Function Documentation

◆ build()

std::unique_ptr< Kernel > luci_interpreter::KernelBuilder::build ( const luci::CircleNode * node )

Definition at line 121 of file KernelBuilder.cpp.

{
  auto specific_builder = _builder_registry->get_kernel_builder_func(node->opcode());
  if (specific_builder != nullptr)
    return specific_builder(node, *this);
 
  std::string msg = "Unsupported operator: ";
  msg += toString(node->opcode()) + " in " + std::string(node->name());
  throw std::invalid_argument(msg.c_str());
}

References luci::must_cast(), luci::CircleNode::name(), and luci::CircleNode::opcode().

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