ONE - On-device Neural Engine
Loading...
Searching...
No Matches
coco::FeatureShape Class Reference

The shape of a feature map. More...

#include <FeatureShape.h>

Collaboration diagram for coco::FeatureShape:

Public Member Functions

 FeatureShape (uint32_t depth, uint32_t height, uint32_t width)
 
 FeatureShape (uint32_t batch, uint32_t depth, uint32_t height, uint32_t width)
 
 FeatureShape (const nncc::core::ADT::feature::Shape &shape)
 
uint32_t batch (void) const
 
- Public Member Functions inherited from nncc::core::ADT::feature::Shape
 Shape (uint32_t depth, uint32_t height, uint32_t width)
 
uint32_t depth (void) const
 
uint32_t height (void) const
 
uint32_t width (void) const
 

Detailed Description

The shape of a feature map.

TODO Implement coco's own FeatureShape without "nncc::core::ADT::feature::Shape"

Definition at line 30 of file FeatureShape.h.

Constructor & Destructor Documentation

◆ FeatureShape() [1/3]

coco::FeatureShape::FeatureShape ( uint32_t  depth,
uint32_t  height,
uint32_t  width 
)
inline

Definition at line 33 of file FeatureShape.h.

34 : Shape{depth, height, width}, _batch{1}
35 {
36 // DO NOTHING
37 }
uint32_t depth(void) const
Definition Shape.h:44
uint32_t width(void) const
Definition Shape.h:46
uint32_t height(void) const
Definition Shape.h:45
Definition Shape.h:28

◆ FeatureShape() [2/3]

coco::FeatureShape::FeatureShape ( uint32_t  batch,
uint32_t  depth,
uint32_t  height,
uint32_t  width 
)
inline

Definition at line 39 of file FeatureShape.h.

40 : Shape{depth, height, width}, _batch{batch}
41 {
42 // DO NOTHING
43 }
uint32_t batch(void) const

◆ FeatureShape() [3/3]

coco::FeatureShape::FeatureShape ( const nncc::core::ADT::feature::Shape shape)
inline

Definition at line 45 of file FeatureShape.h.

45 : Shape{shape}, _batch{1}
46 {
47 // DO NOTHING
48 }

Member Function Documentation

◆ batch()

uint32_t coco::FeatureShape::batch ( void  ) const
inline

Definition at line 51 of file FeatureShape.h.

51{ return _batch; }

Referenced by coco::FeatureLayout::batch().


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