ONE - On-device Neural Engine
|
#include <OutputManager.h>
Public Member Functions | |
OutputManager (Module *m=nullptr) | |
Output * | create (const nncc::core::ADT::tensor::Shape &) |
Public Member Functions inherited from coco::PtrManager< Output > | |
uint32_t | size (void) const |
Return the number of managed objects. | |
Output * | at (uint32_t n) const |
Public Member Functions inherited from coco::EntityBuilder | |
virtual | ~EntityBuilder ()=default |
Additional Inherited Members | |
Protected Member Functions inherited from coco::PtrManager< Output > | |
U * | take (std::unique_ptr< U > &&o) |
std::unique_ptr< Output > | release (Output *ptr) |
Protected Member Functions inherited from coco::EntityBuilder | |
Module * | module (void) const |
void | module (Module *m) |
void | modulize (Entity *entity) const |
Definition at line 28 of file OutputManager.h.
|
inline |
Output * coco::OutputManager::create | ( | const nncc::core::ADT::tensor::Shape & | shape | ) |
Definition at line 24 of file OutputManager.cpp.
References coco::EntityBuilder::modulize(), and coco::PtrManager< Output >::take().