|
ONE - On-device Neural Engine
|
Class to calls custom operation. More...
#include <COpCall.h>

Public Member Functions | |
| COpCall (unsigned arity) | |
| void | op (const std::string &op) |
| const std::string & | op () |
| void | name (const std::string &name) |
| const std::string & | name () |
| void | input (uint32_t nth, loco::Node *node) |
| loco::Node * | input (uint32_t nth) const |
| void | attr (const std::string &attr_name, std::unique_ptr< COpAttrData > &&attr_data) |
| Store [attr_name, attr_data]. | |
| template<COpAttrType AT> | |
| const AttrTypeTrait< AT >::Type * | attr (const std::string &attr_name) const |
| Retrieve attr_data stored with attr_name. | |
| std::vector< std::string > | attr_names () const |
| get all the names of attr | |
Public Member Functions inherited from locoex::VariadicArityNode< COpNode > | |
| VariadicArityNode (uint32_t arity) | |
| virtual | ~VariadicArityNode ()=default |
| uint32_t | arity (void) const final |
| loco::Node * | arg (uint32_t n) const final |
| void | drop (void) final |
Public Member Functions inherited from locoex::COpNode | |
| virtual | ~COpNode ()=default |
| const loco::Dialect * | dialect (void) const final |
| Return "Dialect" identifier that this node belongs to. | |
| uint32_t | opnum (void) const final |
Public Member Functions inherited from loco::Node | |
| Node ()=default | |
| Node (const Node &)=delete | |
| Node (Node &&)=delete | |
| virtual | ~Node () |
| Graph * | graph (void) |
| const Graph * | graph (void) const |
Public Member Functions inherited from loco::AnnotatedItem< NodeAnnotation > | |
| AnnotatedItem ()=default | |
| virtual | ~AnnotatedItem ()=default |
| const T * | annot (void) const |
| Retrieve a stored annotation of type T. | |
| void | annot (std::unique_ptr< T > &&p) |
| Attach or remove a new annotation of type T. | |
Additional Inherited Members | |
Protected Member Functions inherited from locoex::VariadicArityNode< COpNode > | |
| loco::Use * | at (uint32_t n) const |
|
inline |
Definition at line 40 of file COpCall.h.
| const AttrTypeTrait< AT >::Type * locoex::COpCall::attr | ( | const std::string & | attr_name | ) | const |
Retrieve attr_data stored with attr_name.
Definition at line 25 of file COpCall.cpp.
| void locoex::COpCall::attr | ( | const std::string & | attr_name, |
| std::unique_ptr< COpAttrData > && | attr_data | ||
| ) |
Store [attr_name, attr_data].
Definition at line 38 of file COpCall.cpp.
| std::vector< std::string > locoex::COpCall::attr_names | ( | ) | const |
get all the names of attr
Definition at line 46 of file COpCall.cpp.
References attr_names().
Referenced by attr_names().
|
inline |
Definition at line 50 of file COpCall.h.
References locoex::VariadicArityNode< COpNode >::at(), and loco::Use::node().
|
inline |
Definition at line 49 of file COpCall.h.
References locoex::VariadicArityNode< COpNode >::at(), and loco::Use::node().
|
inline |
|
inline |
|
inline |
|
inline |
Definition at line 43 of file COpCall.h.
References op().
Referenced by moco::tf::COpCallGraphBuilder::build().