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