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

Convert TFConv2D to Canonical Conv2D. More...

#include <Conv2DCanonicalizer.h>

Collaboration diagram for moco::tf::Conv2DCanonicalizer:

Public Member Functions

const char * name (void) const final
 
bool transform (TFConv2D *) const final
 
- Public Member Functions inherited from moco::tf::SimpleNodeTransform< TFConv2D >
 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 TFConv2D to Canonical Conv2D.

Definition at line 35 of file Conv2DCanonicalizer.h.

Member Function Documentation

◆ name()

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

Reimplemented from logo::Pass.

Definition at line 38 of file Conv2DCanonicalizer.h.

38{ return "Conv2DCanonicalizer"; }

◆ transform()

bool moco::tf::Conv2DCanonicalizer::transform ( TFConv2D node) const
finalvirtual

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

Definition at line 126 of file Conv2DCanonicalizer.cpp.

127{
128 return canonicalize_conv2d(node->graph(), node);
129}

References loco::Node::graph().


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