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

Public Member Functions

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

Detailed Description

Definition at line 516 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

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

Reimplemented from circledump::OpPrinter.

Definition at line 519 of file OpPrinter.cpp.

520 {
521 if (auto *softmax_params = op->builtin_options_as_SoftmaxOptions())
522 {
523 os << " ";
524 os << "Beta(" << softmax_params->beta() << ")";
525 os << std::endl;
526 }
527 }

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