ONE - On-device Neural Engine
|
Class to Forward send Reshape after UnaryOp. More...
#include <ForwardReshapeToUnaryOpPass.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 Forward send Reshape after UnaryOp.
Definition at line 28 of file ForwardReshapeToUnaryOpPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 30 of file ForwardReshapeToUnaryOpPass.h.
|
finalvirtual |
BEFORE | [CircleNode] [CircleConst] | / [CircleReshape] / | [CircleNode] [(UnaryOp)] | | \ | | [CircleNode] | | |
UnaryOp: CircleNeg, ... Note: Binary Op (Div, Mul) can also be considered as a unary operation if one of its inputs is a constant. For CircleMean in which the axis is a scalar constant and reshape Op does not change the axis on which the mean is taken, the Reshape Op can be forwarded.
AFTER | [CircleConst] [CircleNode] | / | [CircleReshape] [(UnaryOp)] [CircleConst] | | / [CircleNode] [CircleReshape] | | \ | | [CircleNode] | | |
Note: new [CircleReshape] after [(UnaryOp)] added
Implements logo::Pass.
Definition at line 436 of file ForwardReshapeToUnaryOpPass.cpp.
References loco::active_nodes(), and loco::output_nodes().
Referenced by package.infer.session::inference().