tfldump::ReshapePrinter Class Reference

Collaboration diagram for tfldump::ReshapePrinter:

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

Detailed Description

Definition at line 198 of file OpPrinter.cpp.

Member Function Documentation

options()

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

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 201 of file OpPrinter.cpp.

  {
    if (auto *reshape_params = op->builtin_options_as_ReshapeOptions())
    {
      auto new_shape = tflread::as_index_vector(reshape_params->new_shape());
      os << "    ";
      os << "NewShape(" << new_shape << ")";
      os << std::endl;
    }
  }

References tflread::as_index_vector().

---

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

• compiler/tfldump/src/OpPrinter.cpp