ONE - On-device Neural Engine
|
Use the input of "Forward" node instead. More...
#include <RemoveForwardNodePass.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 |
Use the input of "Forward" node instead.
BEFORE: [X] -> [Forward] -> [Y]
AFTER: [X] -> [Y] [Forward]
NOTE This transform does not remove "Forward" node
Definition at line 37 of file RemoveForwardNodePass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 39 of file RemoveForwardNodePass.h.
|
finalvirtual |
Run the pass.
Implements logo::Pass.
Definition at line 27 of file RemoveForwardNodePass.cpp.
References loco::all_nodes(), loco::CanonicalDialect::get(), loco::Forward::input(), and loco::Subst< SubstQualifier::Default >::with().
Referenced by package.infer.session::inference().