#include <CircleExporterUtils.h>

Collaboration diagram for exo::circle_detail::SerializedModelData:

Public Member Functions
	SerializedModelData ()=default

	SerializedModelData (const SerializedModelData &)=delete

uint32_t	registerBuiltinOpcode (circle::BuiltinOperator builtin_code)
	if opcode is not registered in table of opcodes add it

uint32_t	registerCustomOpcode (const std::string &custom_op)

Data Fields
std::unordered_map< OpCode, uint32_t >	_operator_codes

std::unordered_map< OpCode, std::string >	_custom_operator_codes

std::vector< flatbuffers::Offset< circle::Operator > >	_operators

std::vector< flatbuffers::Offset< circle::Tensor > >	_tensors

std::vector< flatbuffers::Offset< circle::Buffer > >	_buffers

std::unordered_map< loco::Pull *, std::string >	_pull_to_name

std::unordered_map< loco::Push *, std::string >	_push_to_name

Data Fields inherited from exo::circle_detail::SubGraphContext
std::vector< int32_t >	_inputs
	SubGraph input tensor id.

std::vector< int32_t >	_outputs
	SubGraph output tensor id.

circle::DataFormat	_data_format {circle::DataFormat::DataFormat_CHANNELS_LAST}
	@DataFormat for SubGraph

Detailed Description

Definition at line 81 of file CircleExporterUtils.h.

Constructor & Destructor Documentation

◆ SerializedModelData() [1/2]

exo::circle_detail::SerializedModelData::SerializedModelData ( )

default

◆ SerializedModelData() [2/2]

exo::circle_detail::SerializedModelData::SerializedModelData ( const SerializedModelData & )

delete

Member Function Documentation

◆ registerBuiltinOpcode()

uint32_t exo::circle_detail::SerializedModelData::registerBuiltinOpcode ( circle::BuiltinOperator builtin_code )

if opcode is not registered in table of opcodes add it

Parameters

builtin_code

Returns: idx of opcode in table of opcodes (see schema)

Definition at line 46 of file CircleExporterUtils.cpp.

{
  auto it = _operator_codes.find(OpCode{builtin_code});
  if (it != _operator_codes.end())
  {
    return it->second;
  }
  auto idx = static_cast<uint32_t>(_operator_codes.size());
  _operator_codes.emplace(OpCode{builtin_code}, idx);
  return idx;
}

References _operator_codes.

Referenced by registerCustomOpcode().

◆ registerCustomOpcode()

uint32_t exo::circle_detail::SerializedModelData::registerCustomOpcode ( const std::string & custom_op )

Definition at line 58 of file CircleExporterUtils.cpp.

{
  circle::BuiltinOperator custom_code = circle::BuiltinOperator_CUSTOM;
  auto idx = registerBuiltinOpcode(custom_code);
  _custom_operator_codes.emplace(OpCode{custom_code}, custom_op);
  return idx;
}

References _custom_operator_codes, and registerBuiltinOpcode().

Field Documentation

◆ _buffers

std::vector<flatbuffers::Offset<circle::Buffer> > exo::circle_detail::SerializedModelData::_buffers

Definition at line 90 of file CircleExporterUtils.h.

Referenced by exo::circle_detail::exportOpDefinedTensor(), and exo::circle_detail::exportOpDefinedTensors().

◆ _custom_operator_codes

std::unordered_map<OpCode, std::string> exo::circle_detail::SerializedModelData::_custom_operator_codes

Definition at line 87 of file CircleExporterUtils.h.

Referenced by registerCustomOpcode().

◆ _operator_codes

std::unordered_map<OpCode, uint32_t> exo::circle_detail::SerializedModelData::_operator_codes

Definition at line 86 of file CircleExporterUtils.h.

Referenced by registerBuiltinOpcode().

◆ _operators

std::vector<flatbuffers::Offset<circle::Operator> > exo::circle_detail::SerializedModelData::_operators

Definition at line 88 of file CircleExporterUtils.h.

◆ _pull_to_name

std::unordered_map<loco::Pull *, std::string> exo::circle_detail::SerializedModelData::_pull_to_name

Definition at line 93 of file CircleExporterUtils.h.

Referenced by exo::circle_detail::registerGraphIOName().

◆ _push_to_name

std::unordered_map<loco::Push *, std::string> exo::circle_detail::SerializedModelData::_push_to_name

Definition at line 94 of file CircleExporterUtils.h.

Referenced by exo::circle_detail::registerGraphIOName().

◆ _tensors

std::vector<flatbuffers::Offset<circle::Tensor> > exo::circle_detail::SerializedModelData::_tensors

Definition at line 89 of file CircleExporterUtils.h.

Referenced by exo::circle_detail::exportOpDefinedTensor().

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

compiler/exo/src/Circle/CircleExporterUtils.h
compiler/exo/src/Circle/CircleExporterUtils.cpp

Public Member Functions

Data Fields