ONE - On-device Neural Engine
|
Class to resolve certain custom op of subgraph into batchmatmul op in circle schema. More...
#include <ResolveCustomOpBatchMatMulPass.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 resolve certain custom op of subgraph into batchmatmul op in circle schema.
Definition at line 28 of file ResolveCustomOpBatchMatMulPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 30 of file ResolveCustomOpBatchMatMulPass.h.
|
finalvirtual |
BEFORE | | [CircleNode] [CircleNode] \ / [CircleCustom]("BatchMatMulV2") | [CircleCustomOut] | [CircleNode] |
AFTER | | [CircleNode] [CircleNode] \ / [CircleBatchMatMul] | [CircleNode] |
Implements logo::Pass.
Definition at line 85 of file ResolveCustomOpBatchMatMulPass.cpp.
References loco::active_nodes(), and loco::output_nodes().
Referenced by package.infer.session::inference().