tfldump::ArgMaxPrinter Class Reference

Collaboration diagram for tfldump::ArgMaxPrinter:

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

Detailed Description

Definition at line 49 of file OpPrinter.cpp.

Member Function Documentation

options()

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

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 52 of file OpPrinter.cpp.

  {
    if (auto *params = op->builtin_options_as_ArgMaxOptions())
    {
      os << "    ";
      os << "OutputType(" << EnumNameTensorType(params->output_type()) << ") ";
      os << std::endl;
    }
  }

---

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

• compiler/tfldump/src/OpPrinter.cpp