ONE - On-device Neural Engine
|
Canonicalize TF-dialect TFMean into canonical TensorReduce(Mean) node. More...
#include <MeanCanonicalizer.h>
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 |
Canonicalize TF-dialect TFMean into canonical TensorReduce(Mean) node.
Definition at line 35 of file MeanCanonicalizer.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 38 of file MeanCanonicalizer.h.
|
finalvirtual |
Implements moco::tf::SimpleNodeTransform< moco::TFMean >.
Definition at line 25 of file MeanCanonicalizer.cpp.