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().