ONE - On-device Neural Engine
|
A KernelLayout connectes each kernel index to an element (in a bag) More...
#include <KernelLayout.h>
Data Structures | |
struct | ID |
Public Member Functions | |
virtual | ~KernelLayout ()=default |
virtual const ID * | id (void) const =0 |
Return the identifier of each layout. | |
virtual const nncc::core::ADT::kernel::Shape & | shape (void) const =0 |
virtual ElemID | at (uint32_t n, uint32_t ch, uint32_t row, uint32_t col) const =0 |
A KernelLayout connectes each kernel index to an element (in a bag)
NOTE KernelLayout is an immutable interface
Definition at line 32 of file KernelLayout.h.
|
virtualdefault |
|
pure virtual |
Implemented in coco::KernelLayouts::NCHW, coco::KernelLayouts::NHWC, and coco::KernelLayouts::Generic.
|
pure virtual |
Return the identifier of each layout.
REQUIRED
Given l1 and l2 of KernelLayout * type, typeid(*l1) == typeif(*l2) SHOULD hold if l1->id() == l2->id() holds.
Implemented in coco::KernelLayouts::NCHW, coco::KernelLayouts::NHWC, and coco::KernelLayouts::Generic.
|
pure virtual |