tfldump::CustomOpPrinter Class Reference

Collaboration diagram for tfldump::CustomOpPrinter:

Public Member Functions
void	options (const tflite::Operator *op, std::ostream &os) const override

Detailed Description

Definition at line 701 of file OpPrinter.cpp.

Member Function Documentation

options()

void tfldump::CustomOpPrinter::options ( const tflite::Operator *  op, std::ostream &  os  ) const

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 704 of file OpPrinter.cpp.

  {
    if (op->custom_options_format() != tflite::CustomOptionsFormat::CustomOptionsFormat_FLEXBUFFERS)
    {
      os << "    ";
      os << "Unknown custom option format";
      return;
    }
 
    const flatbuffers::Vector<uint8_t> *option_buf = op->custom_options();
 
    if (option_buf == nullptr || option_buf->size() == 0)
    {
      os << "No attrs found." << std::endl;
      return;
    }
 
    // printing attrs
    // attrs of custom ops are encoded in flexbuffer format
    auto attr_map = flexbuffers::GetRoot(option_buf->data(), option_buf->size()).AsMap();
 
    os << "    ";
    auto keys = attr_map.Keys();
    for (int i = 0; i < keys.size(); i++)
    {
      auto key = keys[i].ToString();
      os << key << "(" << attr_map[key].ToString() << ") ";
    }
 
    // Note: attr in "Shape" type does not seem to be converted by tflite_convert.
    // When the converted tflite file (with custom op) is opened with hexa editory,
    // attrs names can be found but attr name in "Shape" type is not found.
 
    os << std::endl;
  }

References flexbuffers::Reference::AsMap(), flatbuffers::Vector< T >::data(), flexbuffers::GetRoot(), flexbuffers::Map::Keys(), and flatbuffers::Vector< T >::size().

---

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

• compiler/tfldump/src/OpPrinter.cpp