ONE - On-device Neural Engine
|
Class to fuse TFLRelu or TFLRelu6 into the TensorFlow Lite ops below: More...
#include <FuseReluPass.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 TFLRelu or TFLRelu6 into the TensorFlow Lite ops below:
ADD, AVERAGE_POOL_2D, CONCATENATION, CONV_2D, DEPTHWISE_CONV_2D, FULLY_CONNECTED, L2_NORMALIZATION, L2_POOL_2D, MAX_POOL_2D, MUL
Definition at line 31 of file FuseReluPass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 33 of file FuseReluPass.h.
|
finalvirtual |
Run the pass.
Implements logo::Pass.
Definition at line 92 of file FuseReluPass.cpp.
References loco::active_nodes(), loco::Node::dialect(), locoex::TFLDialect::get(), and loco::output_nodes().
Referenced by package.infer.session::inference().