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

Public Member Functions

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

Detailed Description

Definition at line 147 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

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

Reimplemented from tfldump::OpPrinter.

Definition at line 150 of file OpPrinter.cpp.

151 {
152 if (auto pool_params = op->builtin_options_as_Pool2DOptions())
153 {
154 os << " ";
155 os << "Padding(" << EnumNamePadding(pool_params->padding()) << ") ";
156 os << "Stride.W(" << pool_params->stride_w() << ") ";
157 os << "Stride.H(" << pool_params->stride_h() << ") ";
158 os << "Filter.W(" << pool_params->filter_width() << ") ";
159 os << "Filter.H(" << pool_params->filter_height() << ") ";
160 os << "Activation("
161 << EnumNameActivationFunctionType(pool_params->fused_activation_function()) << ")";
162 os << std::endl;
163 }
164 }

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