Searched refs:handleNegativeAxis (Results 1 – 12 of 12) sorted by relevance
| /packages/modules/NeuralNetworks/common/operations/ |
| D | L2Normalization.cpp | 142 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normFloat32()
171 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8()
187 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in l2normQuant8Signed()
266 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | ChannelShuffle.cpp | 86 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
95 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | Gather.cpp | 85 NN_RET_CHECK(handleNegativeAxis(input, &axis)); in prepare()
103 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | LocalResponseNormalization.cpp | 90 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in localResponseNorm()
128 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in executeTyped()
|
| D | ExpandDims.cpp | 28 NN_CHECK(handleNegativeAxis(getNumberOfDimensions(input) + 1, &axis)); in prepare()
|
| D | Softmax.cpp | 88 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxFloat32()
206 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in softmaxQuant8()
|
| D | ArgMinMax.cpp | 54 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in argMinMaxGeneric()
|
| D | Split.cpp | 32 NN_CHECK(handleNegativeAxis(inputShape, &axis)); in splitGeneric()
|
| D | LogSoftmax.cpp | 99 NN_RET_CHECK(handleNegativeAxis(context->getInputShape(kInputTensor), &axis)); in execute()
|
| D | Reduce.cpp | 142 NN_RET_CHECK(handleNegativeAxis(inputRank, &axis)); in prepare()
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | OperationsUtils.h | 142 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis);
144 inline bool handleNegativeAxis(const Shape& shape, int32_t* axis) { in handleNegativeAxis() function
145 return handleNegativeAxis(getNumberOfDimensions(shape), axis); in handleNegativeAxis()
|
| /packages/modules/NeuralNetworks/common/ |
| D | OperationsUtils.cpp | 167 bool handleNegativeAxis(int32_t numberOfDimensions, int32_t* axis) { in handleNegativeAxis() function
670 NN_CHECK(handleNegativeAxis(input, &axis)); in argMinMaxPrepare()
691 NN_CHECK(handleNegativeAxis(input, &axis)); in splitPrepare()
|