#include <TFLExporterUtils.h>

Collaboration diagram for exo::tflite_detail::SerializedModelData:

Public Member Functions
	SerializedModelData ()=default

	SerializedModelData (const SerializedModelData &)=delete

uint32_t	registerBuiltinOpcode (tflite::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< tflite::Operator > >	_operators

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

std::vector< flatbuffers::Offset< tflite::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::tflite_detail::SubGraphContext
std::vector< int32_t >	_inputs
	SubGraph input tensor id.

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

Detailed Description

Definition at line 79 of file TFLExporterUtils.h.

Constructor & Destructor Documentation

◆ SerializedModelData() [1/2]

exo::tflite_detail::SerializedModelData::SerializedModelData ( )

default

◆ SerializedModelData() [2/2]

exo::tflite_detail::SerializedModelData::SerializedModelData ( const SerializedModelData & )

delete

Member Function Documentation

◆ registerBuiltinOpcode()

uint32_t exo::tflite_detail::SerializedModelData::registerBuiltinOpcode ( tflite::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 TFLExporterUtils.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::tflite_detail::SerializedModelData::registerCustomOpcode ( const std::string & custom_op )

Definition at line 58 of file TFLExporterUtils.cpp.

{
  tflite::BuiltinOperator custom_code = tflite::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<tflite::Buffer> > exo::tflite_detail::SerializedModelData::_buffers

Definition at line 88 of file TFLExporterUtils.h.

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

◆ _custom_operator_codes

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

Definition at line 85 of file TFLExporterUtils.h.

Referenced by registerCustomOpcode().

◆ _operator_codes

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

Definition at line 84 of file TFLExporterUtils.h.

Referenced by registerBuiltinOpcode().

◆ _operators

std::vector<flatbuffers::Offset<tflite::Operator> > exo::tflite_detail::SerializedModelData::_operators

Definition at line 86 of file TFLExporterUtils.h.

◆ _pull_to_name

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

Definition at line 91 of file TFLExporterUtils.h.

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

◆ _push_to_name

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

Definition at line 92 of file TFLExporterUtils.h.

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

◆ _tensors

std::vector<flatbuffers::Offset<tflite::Tensor> > exo::tflite_detail::SerializedModelData::_tensors

Definition at line 87 of file TFLExporterUtils.h.

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

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

compiler/exo/src/TFLite/TFLExporterUtils.h
compiler/exo/src/TFLite/TFLExporterUtils.cpp

Public Member Functions

Data Fields