tfldump::SqueezePrinter Class Reference

Collaboration diagram for tfldump::SqueezePrinter:

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

Detailed Description

Definition at line 536 of file OpPrinter.cpp.

Member Function Documentation

options()

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

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 539 of file OpPrinter.cpp.

  {
    if (auto *params = op->builtin_options_as_SqueezeOptions())
    {
      os << "    ";
      os << "SqueezeDims(";
      for (int i = 0; i < params->squeeze_dims()->size(); ++i)
      {
        if (i != 0)
          os << ", ";
        os << params->squeeze_dims()->Get(i);
      }
      os << ")";
      os << std::endl;
    }
  }

References size.

---

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

• compiler/tfldump/src/OpPrinter.cpp