ONE - On-device Neural Engine
Loading...
Searching...
No Matches
moco::tf::MaxPoolCanonicalizer Class Reference

Convert TFMaxPool to Canonical MaxPool2D. More...

#include <MaxPoolCanonicalizer.h>

Collaboration diagram for moco::tf::MaxPoolCanonicalizer:

Public Member Functions

const char * name (void) const final
 
bool transform (moco::TFMaxPool *) const final
 
- Public Member Functions inherited from moco::tf::SimpleNodeTransform< moco::TFMaxPool >
 SimpleNodeTransform ()=default
 
virtual ~SimpleNodeTransform ()=default
 
bool run (loco::Graph *graph) final
 Run the pass.
 
- Public Member Functions inherited from logo::Pass
virtual ~Pass ()=default
 

Detailed Description

Convert TFMaxPool to Canonical MaxPool2D.

Definition at line 35 of file MaxPoolCanonicalizer.h.

Member Function Documentation

◆ name()

const char * moco::tf::MaxPoolCanonicalizer::name ( void  ) const
inlinefinalvirtual

Reimplemented from logo::Pass.

Definition at line 38 of file MaxPoolCanonicalizer.h.

38{ return "MaxPoolCanonicalizer"; }

◆ transform()

bool moco::tf::MaxPoolCanonicalizer::transform ( moco::TFMaxPool node) const
finalvirtual

Implements moco::tf::SimpleNodeTransform< moco::TFMaxPool >.

Definition at line 105 of file MaxPoolCanonicalizer.cpp.

106{
107 return canonicalize_maxpool2d(node->graph(), node);
108}
Graph * graph(void)
Definition Node.h:70

References loco::Node::graph().


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