|
ONE - On-device Neural Engine
|
Class to fuse certain pattern of subgraph into CircleRoPE with auxiliary nodes. More...
#include <FuseRoPEPass.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 certain pattern of subgraph into CircleRoPE with auxiliary nodes.
For detailed subgraph pattern to be fused, please check its implementation.
Definition at line 31 of file FuseRoPEPass.h.
Reimplemented from logo::Pass.
Definition at line 33 of file FuseRoPEPass.h.
|
finalvirtual |
Run the pass.
Implements logo::Pass.
Definition at line 255 of file FuseRoPEPass.cpp.
References loco::active_nodes(), luci::must_cast(), and loco::output_nodes().