Searched refs:handleNegativeAxis (Results 1 – 12 of 12) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | ChannelShuffle.cpp | 78 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
87 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ExpandDims.cpp | 28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
|
| D | L2Normalization.cpp | 101 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
130 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
194 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | Gather.cpp | 80 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ArgMinMax.cpp | 59 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
|
| D | Split.cpp | 31 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
|
| D | Normalization.cpp | 78 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNormFloat32()
|
| D | Softmax.cpp | 77 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
189 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
|
| D | LogSoftmax.cpp | 99 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | Reduce.cpp | 116 NN_RET_CHECK(handleNegativeAxis(inputRank, &axis)); in prepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 157 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis);
159 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis() function
160 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 145 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis) { in handleNegativeAxis() function
831 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
850 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
|