tfldump::CastPrinter Class Reference

Collaboration diagram for tfldump::CastPrinter:

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

Detailed Description

Definition at line 97 of file OpPrinter.cpp.

Member Function Documentation

options()

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

inlineoverridevirtual

Reimplemented from tfldump::OpPrinter.

Definition at line 100 of file OpPrinter.cpp.

  {
    if (auto cast_params = op->builtin_options_as_CastOptions())
    {
      os << "    ";
      os << "in_data_type(" << tflite::EnumNameTensorType(cast_params->in_data_type()) << ") ";
      os << "out_data_type(" << tflite::EnumNameTensorType(cast_params->out_data_type()) << ") ";
      os << std::endl;
    }
  }

---

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

• compiler/tfldump/src/OpPrinter.cpp