ONE - On-device Neural Engine
|
Class to remove unnecessary Reshape nodes. More...
#include <RemoveUnnecessaryReshapeNetPass.h>
Public Member Functions | |
const char * | name (void) const final |
bool | run (loco::Graph *g) final |
Public Member Functions inherited from logo::Pass | |
virtual | ~Pass ()=default |
Class to remove unnecessary Reshape nodes.
This class will remove unnecessary pre/post-Reshape nodes. See https://github.com/Samsung/ONE/issues/9600 for more details.
Definition at line 30 of file RemoveUnnecessaryReshapeNetPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 32 of file RemoveUnnecessaryReshapeNetPass.h.
|
finalvirtual |
BEFORE
[CircleNode] |
[CircleReshape_1] (shape: A -> B) | [CircleNode] (ex: Add/Mul/Tanh/Logistic ..) | [CircleReshape_2] (shape: B -> A) | [CircleNode]
AFTER
[CircleNode] | \ | [CircleReshape_1] [CircleNode] | \ | [CircleReshape_2] [CircleNode]
Implements logo::Pass.
Definition at line 159 of file RemoveUnnecessaryReshapeNetPass.cpp.
References loco::active_nodes(), and loco::output_nodes().
Referenced by package.infer.session::inference().