ONE - On-device Neural Engine
|
A wrapper class for an integer number that specifies axis. More...
#include <ShapeQuery.h>
Public Member Functions | |
AxisSpecifier (int32_t value) | |
int32_t | value (void) const |
A wrapper class for an integer number that specifies axis.
Several Caffe layers includes 'axis' parameter (which may be negative) which specifies some axis required for operation.
Here are several examples:
AxisSpecifier class is introduced to distinguish this 'axis' parameter from other integers (to prevent possible mistake).
Definition at line 37 of file ShapeQuery.h.
|
inlineexplicit |
Definition at line 40 of file ShapeQuery.h.
|
inline |