ONE - On-device Neural Engine
|
#include <Shape.h>
Data Fields | |
OperandType | type |
std::vector< uint32_t > | dimensions |
float | scale |
int32_t | offset |
std::vector<uint32_t> Shape::dimensions |
Definition at line 30 of file Shape.h.
Referenced by addPrepare(), convertShapeToDims(), divPrepare(), getNumberOfDimensions(), getNumberOfElements(), getSizeOfDimension(), mulPrepare(), reshapeGeneric(), SameShape(), SetShape(), softmaxFloat32(), and subPrepare().
OperandType Shape::type |
Definition at line 29 of file Shape.h.
Referenced by addPrepare(), divPrepare(), fullyConnectedPrepare(), mulPrepare(), padGeneric(), padPrepare(), reshapeGeneric(), SameShape(), SetShape(), RunTimeOperandInfo::shape(), and subPrepare().