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

Convert TFConv2DBackpropInput to Canonical TransposedConv2D. More...

#include <Conv2DBackpropInputCanonicalizer.h>

Collaboration diagram for moco::tf::Conv2DBackpropInputCanonicalizer:

Public Member Functions

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

Definition at line 33 of file Conv2DBackpropInputCanonicalizer.h.

Member Function Documentation

◆ name()

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

Reimplemented from logo::Pass.

Definition at line 36 of file Conv2DBackpropInputCanonicalizer.h.

36{ return "Conv2DBackpropInputCanonicalizer"; }

◆ transform()

bool moco::tf::Conv2DBackpropInputCanonicalizer::transform ( moco::TFConv2DBackpropInput node) const
finalvirtual

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

Definition at line 365 of file Conv2DBackpropInputCanonicalizer.cpp.

366{
367 return canonicalize_conv2d_backprop_input(node->graph(), node);
368}
Graph * graph(void)
Definition Node.h:70

References loco::Node::graph().


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