ONE - On-device Neural Engine
|
Class to fuse TFLReshape + TFLConst into one equivalent TFLConst. More...
#include <FoldReshapeOfConstPass.h>
Public Member Functions | |
const char * | name (void) const final |
bool | run (loco::Graph *g) final |
Run the pass. | |
Public Member Functions inherited from logo::Pass | |
virtual | ~Pass ()=default |
Class to fuse TFLReshape + TFLConst into one equivalent TFLConst.
<before> TFLConst — TFLReshape — Out
<after> TFLConst — TFLReshape — TFLConst (new) ---------— Out
TODO This pass is for temporary. Deprecate this pass.
Definition at line 37 of file FoldReshapeOfConstPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 39 of file FoldReshapeOfConstPass.h.
|
finalvirtual |
Run the pass.
Implements logo::Pass.
Definition at line 101 of file FoldReshapeOfConstPass.cpp.
References loco::active_nodes(), and loco::output_nodes().
Referenced by package.infer.session::inference().