ONE - On-device Neural Engine
|
Run shape inference to the graph. More...
#include <ShapeInferencePass.h>
Public Member Functions | |
const char * | name (void) const final |
bool | run (loco::Graph *graph) override |
Run the pass. | |
Public Member Functions inherited from logo::Pass | |
virtual | ~Pass ()=default |
Run shape inference to the graph.
Definition at line 32 of file ShapeInferencePass.h.
|
inlinefinalvirtual |
Reimplemented from logo::Pass.
Definition at line 35 of file ShapeInferencePass.h.
|
overridevirtual |
Run the pass.
Implements logo::Pass.
Definition at line 40 of file ShapeInferencePass.cpp.
References loco::apply(), loco::MultiDialectShapeInferenceRule::bind(), loco::CanonicalDialect::get(), locoex::COpDialect::get(), moco::TFDialect::get(), and loco::ShapeInferenceSession::to().
Referenced by package.infer.session::inference().