Searched refs:handleNegativeAxis (Results 1 – 12 of 12) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | L2Normalization.cpp | 141 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
170 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
186 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8Signed()
261 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ChannelShuffle.cpp | 89 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
98 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ExpandDims.cpp | 28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
|
| D | Gather.cpp | 88 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
106 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ArgMinMax.cpp | 58 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
|
| D | LocalResponseNormalization.cpp | 89 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNorm()
126 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in executeTyped()
|
| D | Split.cpp | 33 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
|
| D | LogSoftmax.cpp | 101 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | Softmax.cpp | 85 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
203 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
|
| D | Reduce.cpp | 140 NN_RET_CHECK(handleNegativeAxis(inputRank, &axis)); in prepare()
|
| /frameworks/ml/nn/common/include/ |
| D | OperationsUtils.h | 158 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis);
160 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis() function
161 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
|
| /frameworks/ml/nn/common/ |
| D | OperationsUtils.cpp | 169 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis) { in handleNegativeAxis() function
676 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
697 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
|