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

Canonicalize TF-dialect TFMean into canonical TensorReduce(Mean) node. More...

#include <MeanCanonicalizer.h>

Collaboration diagram for moco::tf::MeanCanonicalizer:

Public Member Functions

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

Canonicalize TF-dialect TFMean into canonical TensorReduce(Mean) node.

Definition at line 35 of file MeanCanonicalizer.h.

Member Function Documentation

◆ name()

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

Reimplemented from logo::Pass.

Definition at line 38 of file MeanCanonicalizer.h.

38{ return "MeanCanonicalizer"; }

◆ transform()

bool moco::tf::MeanCanonicalizer::transform ( moco::TFMean node) const
finalvirtual

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

Definition at line 25 of file MeanCanonicalizer.cpp.

26{
27 return canonicalize_reduce_node(node);
28}

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