|
ONE - On-device Neural Engine
|
Class to transform Div(X,Sqrt(y)) to Mul(X,Rsqrt(y)) More...
#include <TransformSqrtDivToRsqrtMulPass.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 transform Div(X,Sqrt(y)) to Mul(X,Rsqrt(y))
Definition at line 28 of file TransformSqrtDivToRsqrtMulPass.h.
Reimplemented from logo::Pass.
Definition at line 30 of file TransformSqrtDivToRsqrtMulPass.h.
|
finalvirtual |
Run the pass.
Implements logo::Pass.
Definition at line 85 of file TransformSqrtDivToRsqrtMulPass.cpp.
References loco::active_nodes(), luci::must_cast(), and loco::output_nodes().