|
ONE - On-device Neural Engine
|
Generic Feature Layout. More...
#include <FeatureLayouts.h>

Public Member Functions | |
| const FeatureLayout::ID * | id (void) const override |
| const FeatureShape & | shape (void) const override |
| ElemID & | at (uint32_t b, uint32_t ch, uint32_t row, uint32_t col) |
| ElemID | at (uint32_t b, uint32_t ch, uint32_t row, uint32_t col) const override |
| void | reorder (const nncc::core::ADT::feature::Layout &l) |
Public Member Functions inherited from coco::FeatureLayout | |
| virtual | ~FeatureLayout ()=default |
| uint32_t | batch (void) const |
| uint32_t | depth (void) const |
| uint32_t | height (void) const |
| uint32_t | width (void) const |
Static Public Member Functions | |
| static const FeatureLayout::ID * | uid (void) |
| static std::unique_ptr< Generic > | create (const nncc::core::ADT::feature::Shape &shape) |
Generic Feature Layout.
Definition at line 127 of file FeatureLayouts.h.
| ElemID & coco::FeatureLayouts::Generic::at | ( | uint32_t | b, |
| uint32_t | ch, | ||
| uint32_t | row, | ||
| uint32_t | col | ||
| ) |
Definition at line 178 of file FeatureLayouts.cpp.
Referenced by reorder().
|
overridevirtual |
Implements coco::FeatureLayout.
Definition at line 183 of file FeatureLayouts.cpp.
|
static |
Definition at line 204 of file FeatureLayouts.cpp.
References shape().
|
inlineoverridevirtual |
Implements coco::FeatureLayout.
Definition at line 134 of file FeatureLayouts.h.
References uid().
Referenced by uid().
| void coco::FeatureLayouts::Generic::reorder | ( | const nncc::core::ADT::feature::Layout & | l | ) |
Definition at line 188 of file FeatureLayouts.cpp.
References at(), coco::FeatureLayout::batch(), nncc::core::ADT::feature::Shape::depth(), nncc::core::ADT::feature::Shape::height(), shape(), and nncc::core::ADT::feature::Shape::width().
|
inlineoverridevirtual |
Implements coco::FeatureLayout.
Definition at line 136 of file FeatureLayouts.h.
Referenced by RandomDataGenerator.RandomDataGenerator::_gen_float32(), RandomDataGenerator.RandomDataGenerator::_gen_int16(), RandomDataGenerator.RandomDataGenerator::_gen_uint8(), create(), and reorder().
|
static |
Definition at line 157 of file FeatureLayouts.cpp.
References id().
Referenced by id().