tfldump::OneHotPrinter Class Reference

Collaboration diagram for tfldump::OneHotPrinter:

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

Detailed Description

Definition at line 479 of file OpPrinter.cpp.

Member Function Documentation

options()

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

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 482 of file OpPrinter.cpp.

  {
    if (auto *params = op->builtin_options_as_OneHotOptions())
    {
      os << "    ";
      os << "Axis(" << params->axis() << ") ";
 
      os << std::endl;
    }
  }

---

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

• compiler/tfldump/src/OpPrinter.cpp