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

Public Member Functions

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

Detailed Description

Definition at line 289 of file OpPrinter.cpp.

Member Function Documentation

◆ options()

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

Reimplemented from tfldump::OpPrinter.

Definition at line 292 of file OpPrinter.cpp.

293 {
294 if (auto conv_params = op->builtin_options_as_DepthwiseConv2DOptions())
295 {
296 os << " ";
297 os << "Padding(" << EnumNamePadding(conv_params->padding()) << ") ";
298 os << "Stride.W(" << conv_params->stride_w() << ") ";
299 os << "Stride.H(" << conv_params->stride_h() << ") ";
300 os << "DepthMultiplier(" << conv_params->depth_multiplier() << ") ";
301 os << "Dilation.W(" << conv_params->dilation_w_factor() << ") ";
302 os << "Dilation.H(" << conv_params->dilation_h_factor() << ") ";
303 os << "Activation("
304 << EnumNameActivationFunctionType(conv_params->fused_activation_function()) << ") ";
305 os << std::endl;
306 }
307 }

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