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

Public Member Functions

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

Detailed Description

Definition at line 112 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

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

Reimplemented from tfldump::OpPrinter.

Definition at line 115 of file OpPrinter.cpp.

116 {
117 if (auto conv_params = op->builtin_options_as_Conv2DOptions())
118 {
119 os << " ";
120 os << "Padding(" << EnumNamePadding(conv_params->padding()) << ") ";
121 os << "Stride.W(" << conv_params->stride_w() << ") ";
122 os << "Stride.H(" << conv_params->stride_h() << ") ";
123 os << "Dilation.W(" << conv_params->dilation_w_factor() << ") ";
124 os << "Dilation.H(" << conv_params->dilation_h_factor() << ") ";
125 os << "Activation("
126 << EnumNameActivationFunctionType(conv_params->fused_activation_function()) << ")";
127 os << std::endl;
128 }
129 }

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