ONE - On-device Neural Engine
|
#include <TFPack.h>
Public Member Functions | |
TFPack (uint32_t arity) | |
Node * | values (uint32_t index) const |
void | values (uint32_t index, Node *node) |
uint32_t | N (void) const |
int32_t | axis (void) const |
void | axis (int32_t axis) |
Public Member Functions inherited from moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > > | |
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 moco::TFNodeImpl< TFOpcode::Pack > | |
virtual | ~TFNodeImpl ()=default |
uint32_t | opnum (void) const final |
TFOpcode | opcode (void) const final |
Public Member Functions inherited from moco::TFNode | |
virtual | ~TFNode ()=default |
const loco::Dialect * | dialect (void) const final |
Return "Dialect" identifier that this node belongs to. | |
template<typename T > | |
T | accept (TFNodeVisitorBase< T > *) const |
template<typename T > | |
T | accept (TFNodeMutableVisitorBase< T > *) |
NodeName | name (void) const |
void | name (const NodeName &name) |
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 moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > > | |
loco::Use * | at (uint32_t n) const |
|
inline |
Definition at line 56 of file TFPack.h.
References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity().
|
inline |
|
inline |
|
inline |
Definition at line 75 of file TFPack.h.
References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity().
|
inline |
Definition at line 63 of file TFPack.h.
References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity(), moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::at(), moco::index(), and loco::Use::node().
|
inline |
Definition at line 68 of file TFPack.h.
References moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::arity(), moco::VariadicArityNode< TFNodeImpl< TFOpcode::Pack > >::at(), moco::index(), and loco::Use::node().