ONE - On-device Neural Engine
Loading...
Searching...
No Matches
circledump::ArgMaxPrinter Class Reference
Collaboration diagram for circledump::ArgMaxPrinter:

Public Member Functions

void options (const circle::Operator *op, std::ostream &os) const override
 

Detailed Description

Definition at line 50 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

void circledump::ArgMaxPrinter::options ( const circle::Operator *  op,
std::ostream &  os 
) const
inlineoverridevirtual

Reimplemented from circledump::OpPrinter.

Definition at line 53 of file OpPrinter.cpp.

54 {
55 if (auto *params = op->builtin_options_as_ArgMaxOptions())
56 {
57 os << " ";
58 os << "OutputType(" << EnumNameTensorType(params->output_type()) << ") ";
59 os << std::endl;
60 }
61 }

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