ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Public Member Functions | Friends
loco::NodePool Class Referencefinal

#include <NodePool.h>

Collaboration diagram for loco::NodePool:
Collaboration graph
[legend]

Public Member Functions

 ~NodePool ()
 
template<typename Derived , typename... Args>
Derived * create (Args &&...args)
 
void destroy (Node *node)
 
- Public Member Functions inherited from loco::ObjectPool< Node >
virtual ~ObjectPool ()=default
 
uint32_t size (void) const
 Return the number of objects.
 
Node * at (uint32_t n) const
 Access N-th object.
 

Friends

class Graph
 

Additional Inherited Members

- Protected Member Functions inherited from loco::ObjectPool< Node >
U * take (std::unique_ptr< U > &&o)
 Take the ownership of a given object and returns its raw pointer.
 
bool erase (Node *ptr)
 Erase an object from the pool.
 

Detailed Description

Definition at line 28 of file NodePool.h.

Constructor & Destructor Documentation

◆ ~NodePool()

loco::NodePool::~NodePool ( )

Definition at line 22 of file NodePool.cpp.

23{
24 // Drop all the references before deallocation
25 for (uint32_t n = 0; n < size(); ++n)
26 {
27 at(n)->drop();
28 }
29}
loco::Node::drop
virtual void drop(void)=0
Drop all the reference of arguments.
loco::ObjectPool< Node >::at
Node * at(uint32_t n) const
Access N-th object.
Definition ObjectPool.h:41
loco::ObjectPool< Node >::size
uint32_t size(void) const
Return the number of objects.
Definition ObjectPool.h:38

References loco::ObjectPool< Node >::at(), loco::Node::drop(), and loco::ObjectPool< Node >::size().

Member Function Documentation

◆ create()

template<typename Derived , typename... Args>
Derived * loco::NodePool::create ( Args &&...  args)
inline

Definition at line 37 of file NodePool.h.

38 {
39 std::unique_ptr<Derived> ptr{new Derived(std::forward<Args>(args)...)};
40 ptr->graph(_graph);
41 return ObjectPool<Node>::take<Derived>(std::move(ptr));
42 }

Referenced by luci::GraphBuilder::build(), luci::GraphBuilderMultiOutput::build(), luci::ConstantFoldingAddTestGraph::ConstantFoldingAddTestGraph(), luci::ConstantFoldingTestGraph::ConstantFoldingTestGraph(), exo::TensorConcatConverter::convert(), InputLayer::operator()(), OutputLayer::operator()(), ReLULayer::operator()(), ConstGenLayer::operator()(), FeatureEncodeLayer::operator()(), FeatureDecodeLayer::operator()(), FilterEncodeLayer::operator()(), DepthwiseFilterEncodeLayer::operator()(), DepthwiseConv2DLayer::operator()(), TransposedConv2DLayer::operator()(), FixedReshapeLayer::operator()(), TensorBroadcastLayer::operator()(), luci::FuseBCQPass::run(), locoex::set_new_shape(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), luci::CloneNode::visit(), and luci::CloneNode::visit().

◆ destroy()

void loco::NodePool::destroy ( Node *  node)
inline

Definition at line 44 of file NodePool.h.

45 {
46 if (!ObjectPool<Node>::erase(node))
47 {
48 throw std::invalid_argument{"node"};
49 }
50 }

Friends And Related Symbol Documentation

◆ Graph

friend class Graph
friend

Definition at line 31 of file NodePool.h.

The documentation for this class was generated from the following files: