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

Convert TFConcatV2 to Canonical TensorConcat. More...

#include <ConcatV2Canonicalizer.h>

Collaboration diagram for moco::tf::ConcatV2Canonicalizer:

Public Member Functions

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

Definition at line 35 of file ConcatV2Canonicalizer.h.

Member Function Documentation

◆ name()

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

Reimplemented from logo::Pass.

Definition at line 38 of file ConcatV2Canonicalizer.h.

38{ return "ConcatV2Canonicalizer"; }

◆ transform()

bool moco::tf::ConcatV2Canonicalizer::transform ( moco::TFConcatV2 node) const
finalvirtual

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

Definition at line 153 of file ConcatV2Canonicalizer.cpp.

154{
155 return canonicalize_concat(node->graph(), node);
156}
Graph * graph(void)
Definition Node.h:70

References loco::Node::graph().


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