|
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().