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

Convert TFReshape to Canonical Reshape. More...

#include <ReshapeCanonicalizer.h>

Collaboration diagram for moco::tf::ReshapeCanonicalizer:

Public Member Functions

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

Definition at line 35 of file ReshapeCanonicalizer.h.

Member Function Documentation

◆ name()

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

Reimplemented from logo::Pass.

Definition at line 38 of file ReshapeCanonicalizer.h.

38{ return "ReshapeCanonicalizer"; }

◆ transform()

bool moco::tf::ReshapeCanonicalizer::transform ( moco::TFReshape node) const
finalvirtual

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

Definition at line 163 of file ReshapeCanonicalizer.cpp.

164{
165 return canonicalize_reshape(node->graph(), node);
166}
Graph * graph(void)
Definition Node.h:70

References loco::Node::graph().


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