#include <CircleOperationExporterRule.h>

Collaboration diagram for luci::OperationExporterRule:

[legend]

Public Member Functions
	OperationExporterRule (ExportContext &ctx)

void	visit (luci::CircleNode *node) final
	Default fallback.

void	visit (luci::CircleConst *) final

void	visit (luci::CircleInput *) final

void	visit (luci::CircleOutput *) final

void	visit (luci::CircleOutputDummy *) final

void	visit (luci::CircleOutputExclude *) final

void	visit (luci::CircleBidirectionalSequenceLSTMOut *) final

void	visit (luci::CircleCustomOut *) final

void	visit (luci::CircleIfOut *) final

void	visit (luci::CircleNonMaxSuppressionV4Out *) final

void	visit (luci::CircleNonMaxSuppressionV5Out *) final

void	visit (luci::CircleSplitOut *) final

void	visit (luci::CircleSplitVOut *) final

void	visit (luci::CircleTopKV2Out *) final

void	visit (luci::CircleUniqueOut *) final

void	visit (luci::CircleUnpackOut *) final

void	visit (luci::CircleVariable *) final

void	visit (luci::CircleWhileOut *) final

Public Member Functions inherited from luci::CircleNodeMutableVisitor< void >
virtual	~CircleNodeMutableVisitor ()=default

Public Member Functions inherited from luci::CircleNodeMutableVisitorBase< T >
virtual	~CircleNodeMutableVisitorBase ()=default

Protected Attributes
ExportContext &	_ctx

Detailed Description

Definition at line 36 of file CircleOperationExporterRule.h.

Constructor & Destructor Documentation

◆ OperationExporterRule()

luci::OperationExporterRule::OperationExporterRule ( ExportContext & ctx )

inline

Definition at line 39 of file CircleOperationExporterRule.h.

                                            : _ctx{ctx}
  {
    // DO NOTHING
  }

Member Function Documentation

◆ visit() [1/18]

void luci::OperationExporterRule::visit ( luci::CircleBidirectionalSequenceLSTMOut * )

inlinefinal

Definition at line 57 of file CircleOperationExporterRule.h.

57{}

◆ visit() [2/18]

void luci::OperationExporterRule::visit ( luci::CircleConst * )

inlinefinal

Definition at line 49 of file CircleOperationExporterRule.h.

49{/* skip, everything is done in exportOpDefinedTensors */};

◆ visit() [3/18]

void luci::OperationExporterRule::visit ( luci::CircleCustomOut * )

inlinefinal

Definition at line 58 of file CircleOperationExporterRule.h.

58{}

◆ visit() [4/18]

void luci::OperationExporterRule::visit ( luci::CircleIfOut * )

inlinefinal

Definition at line 59 of file CircleOperationExporterRule.h.

59{}

◆ visit() [5/18]

void luci::OperationExporterRule::visit ( luci::CircleInput * )

inlinefinal

Definition at line 52 of file CircleOperationExporterRule.h.

52{}

◆ visit() [6/18]

void luci::OperationExporterRule::visit ( luci::CircleNode * )

finalvirtual

Default fallback.

Reimplemented from luci::CircleNodeMutableVisitor< void >.

Definition at line 250 of file CircleOperationExporterRule.cpp.

{
  auto op_idx = _ctx.md.registerBuiltinOpcode(circle_builtin_operator(node),
                                              circle_custom_code(node), node->op_version());
 
  std::vector<int32_t> inputs_vec;
  for (uint32_t i = 0; i < node->arity(); ++i)
    inputs_vec.push_back(luci::get_tensor_index(node->arg(i)));
  auto inputs = _ctx.builder.CreateVector(inputs_vec);
 
  OutputVectorExtractor outputs_vec_extractor;
  auto outputs_vec = node->accept(&outputs_vec_extractor);
  auto outputs = _ctx.builder.CreateVector(outputs_vec);
 
  auto builtin_options = circle_builtin_options(node);
 
  luci::BuiltinOptionsExtractor builtin_options_extractor(_ctx.builder);
  auto options_offset = node->accept(&builtin_options_extractor);
 
  // If node is not CircleCustom, null offset(0) is returned
  auto custom_options = circle_custom_options(_ctx.builder, node);
 
  auto op_offset = circle::CreateOperator(_ctx.builder, op_idx, inputs, outputs, builtin_options,
                                          options_offset, custom_options);
  _ctx.gd._operators.push_back(op_offset);
}