Class that holds operator codes and related methods. More...

#include <Context.h>

Public Member Functions
	TflOpCodeContext (const flatbuffers::Vector< flatbuffers::Offset< tflite::OperatorCode > > *opcodes)

tflite::BuiltinOperator	builtin_code (const tflite::Operator *op) const
	Returns BuiltinOperator value of the operator.

std::string	opcode_name (const tflite::Operator *op) const
	Returns human readable name of the operator code of the operator.

Static Public Member Functions
static bool	is_valid (const tflite::OperatorCode *opcode)

static bool	is_custom (const tflite::OperatorCode *opcode)

Detailed Description

Class that holds operator codes and related methods.

Definition at line 56 of file Context.h.

Constructor & Destructor Documentation

◆ TflOpCodeContext()

tflimport::TflOpCodeContext::TflOpCodeContext ( const flatbuffers::Vector< flatbuffers::Offset< tflite::OperatorCode > > * opcodes )

Definition at line 50 of file Context.cpp.

{
  for (const tflite::OperatorCode *opcode : *opcodes)
  {
    _opcodes.push_back(opcode);
  }
}

Member Function Documentation

◆ builtin_code()

tflite::BuiltinOperator tflimport::TflOpCodeContext::builtin_code ( const tflite::Operator * op ) const

Returns BuiltinOperator value of the operator.

Definition at line 59 of file Context.cpp.

{
  uint32_t index = op->opcode_index();
  assert(index < _opcodes.size());
  const tflite::OperatorCode *opcode = _opcodes.at(index);
  return opcode->builtin_code();
}

◆ is_custom()

bool tflimport::TflOpCodeContext::is_custom ( const tflite::OperatorCode * opcode )

static

Definition at line 98 of file Context.cpp.

{
  tflite::BuiltinOperator code = opcode->builtin_code();
  return (code == tflite::BuiltinOperator_CUSTOM);
}

Referenced by opcode_name().

◆ is_valid()

bool tflimport::TflOpCodeContext::is_valid ( const tflite::OperatorCode * opcode )

static

Definition at line 92 of file Context.cpp.

{
  tflite::BuiltinOperator code = opcode->builtin_code();
  return (tflite::BuiltinOperator_MIN <= code && code <= tflite::BuiltinOperator_MAX);
}

Referenced by opcode_name().

◆ opcode_name()

std::string tflimport::TflOpCodeContext::opcode_name ( const tflite::Operator * op ) const

Returns human readable name of the operator code of the operator.

Note: TF lite InterpreterBuilder sets an error state and returns error code for invalid opcode. Here we just return human readable message as this method returns a name for the operator code.

Definition at line 67 of file Context.cpp.

{
  uint32_t index = op->opcode_index();
  assert(index < _opcodes.size());
  const tflite::OperatorCode *opcode = _opcodes.at(index);
 
  if (!is_valid(opcode))
  {
    std::ostringstream oss;
    oss << "(invalid: " << index << ")";
    return oss.str();
  }
 
  if (is_custom(opcode))
  {
    if (!opcode->custom_code())
      return "(invalid custom)";
 
    return opcode->custom_code()->c_str();
  }
 
  tflite::BuiltinOperator code = opcode->builtin_code();
  return EnumNameBuiltinOperator(code);
}

References is_custom(), and is_valid().

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

compiler/enco/frontend/tflite/src/Context.h
compiler/enco/frontend/tflite/src/Context.cpp

Public Member Functions

Static Public Member Functions