ONE - On-device Neural Engine
|
Pass to infer shape of circle nodes. More...
#include <CircleShapeInferencePass.h>
Public Member Functions | |
virtual const char * | name (void) const |
bool | run (luci::Module *m) |
bool | run (loco::Graph *graph) |
Run the pass. | |
Public Member Functions inherited from logo::Pass | |
virtual | ~Pass ()=default |
Pass to infer shape of circle nodes.
Definition at line 30 of file CircleShapeInferencePass.h.
|
inlinevirtual |
Reimplemented from logo::Pass.
Definition at line 33 of file CircleShapeInferencePass.h.
|
virtual |
Run the pass.
Implements logo::Pass.
Definition at line 42 of file CircleShapeInferencePass.cpp.
References loco::TensorShape::dim(), luci::sinf::Rule::infer(), luci::inference_candidates(), luci::is_same_shape(), loco::TensorShape::rank(), and luci::VALID.
Referenced by package.infer.session::inference().
|
virtual |
Implements luci::Pass.
Definition at line 29 of file CircleShapeInferencePass.cpp.
Referenced by package.infer.session::inference(), and run().