ONE - On-device Neural Engine
|
Class to substitute certain SplitV to Split. More...
#include <SubstituteSplitVToSplitPass.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 substitute certain SplitV to Split.
Definition at line 28 of file SubstituteSplitVToSplitPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 30 of file SubstituteSplitVToSplitPass.h.
|
finalvirtual |
EXAMPLE (SplitV with num_split = 2)
BEFORE [CircleNode] | [CircleSplitV] (size_splits and split_dim are ignored) / \ [CircleSplitVOut] [CircleSplitVOut] | | [CircleNode] [CircleNode]
AFTER [CircleNode] / \ [CircleSplit] [CircleSplitV] (dead) / \ \ [CircleSplitOut] [CircleSplitOut] [CircleSplitVOut] * 2 (dead) | | [CircleNode] [CircleNode]
Implements logo::Pass.
Definition at line 107 of file SubstituteSplitVToSplitPass.cpp.
References loco::active_nodes(), and loco::output_nodes().
Referenced by package.infer.session::inference().