ONE - On-device Neural Engine
|
#include <CircleShapeInferenceRule.h>
Public Member Functions | |
bool | recognize (const loco::Dialect *) const final |
Return true if this rule recognizes a given dialect. | |
bool | infer (const loco::Node *, loco::NodeShape &) const final |
Infer node's shape. | |
Public Member Functions inherited from loco::ShapeInferenceRule | |
virtual | ~ShapeInferenceRule ()=default |
virtual bool | support (const API &api) const |
Check whether a given API is available or not. | |
virtual void | infer (const Context *, const Node *, Sink *) const |
Additional Inherited Members | |
Public Types inherited from loco::ShapeInferenceRule | |
enum class | API { V1 , V2 } |
Definition at line 25 of file CircleShapeInferenceRule.h.
|
finalvirtual |
Infer node's shape.
WARNING!!
Implementation SHOULD return true only when it succeeds in inference!
Implements loco::ShapeInferenceRule.
Definition at line 56 of file CircleShapeInferenceRule.cpp.
References loco::Node::dialect(), and locoex::CircleDialect::get().
|
finalvirtual |
Return true if this rule recognizes a given dialect.
Implements loco::ShapeInferenceRule.
Definition at line 51 of file CircleShapeInferenceRule.cpp.
References locoex::CircleDialect::get().