ONE - On-device Neural Engine
|
Data Structures | |
struct | BinaryFunc |
struct | NodeData |
Read-only no-template wrapper for 'Buffer'. Serves interface for input and output of 'Session'. More... | |
class | NodeDataImpl |
An implementation of NodeData interface. More... | |
struct | NodeDomain |
class | NodeExecution |
Helper class for Session, responsible to process one node calculation. More... | |
class | Session |
Session for loco graph inference. More... | |
struct | UnaryFunc |
Functions | |
template<typename DT > | |
std::unique_ptr< NodeData > | make_data (const NodeData::Buffer< DT > &buffer) |
Copy buffer to make NodeData. | |
template<> | |
std::unique_ptr< NodeData > | make_data (const NodeData::Buffer< int32_t > &buf) |
template<> | |
std::unique_ptr< NodeData > | make_data (const NodeData::Buffer< float > &buf) |
void | annot_domain (loco::Node *node, const loco::Domain &domain) |
Wrapper to annotate domain to node. Cannot annotate unknown domain. | |
loco::Domain | annot_domain (const loco::Node *node) |
Wrapper to get domain annotation of node. | |
void | erase_annot_domain (loco::Node *node) |
Erase already annotated node domain. | |
void | validate (bool true_cond, const std::string &&exception_msg) |
loco::Domain locomotiv::annot_domain | ( | const loco::Node * | node | ) |
Wrapper to get domain annotation of node.
Definition at line 42 of file NodeDomain.cpp.
References loco::AnnotatedItem< Annotation >::annot(), loco::Unknown, and locomotiv::NodeDomain::value.
void locomotiv::annot_domain | ( | loco::Node * | node, |
const loco::Domain & | domain | ||
) |
Wrapper to annotate domain to node. Cannot annotate unknown domain.
Definition at line 34 of file NodeDomain.cpp.
References loco::AnnotatedItem< Annotation >::annot(), and loco::Unknown.
void locomotiv::erase_annot_domain | ( | loco::Node * | node | ) |
Erase already annotated node domain.
Definition at line 51 of file NodeDomain.cpp.
References loco::AnnotatedItem< Annotation >::annot().
Referenced by locomotiv::Session::~Session().
std::unique_ptr< NodeData > locomotiv::make_data | ( | const NodeData::Buffer< DT > & | buffer | ) |
Copy buffer to make NodeData.
References make_data().
Referenced by make_data(), make_data(), make_data(), and nnkit::support::moco::tf::Backend::prepare().
std::unique_ptr< NodeData > locomotiv::make_data | ( | const NodeData::Buffer< float > & | buf | ) |
std::unique_ptr< NodeData > locomotiv::make_data | ( | const NodeData::Buffer< int32_t > & | buf | ) |
Definition at line 23 of file NodeData.cpp.
References make_data().
|
inline |
Definition at line 26 of file Validation.h.