Searched refs:outerSize (Results 1 – 9 of 9) sorted by relevance
/packages/modules/NeuralNetworks/common/operations/ |
D | L2Normalization.cpp | 52 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl() local 56 for (uint32_t outer = 0; outer < outerSize; ++outer) { in l2normFloat32Impl() 79 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl() local 83 for (uint32_t outer = 0; outer < outerSize; ++outer) { in l2normQuant8Impl() 111 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8SignedImpl() local 115 for (uint32_t outer = 0; outer < outerSize; ++outer) { in l2normQuant8SignedImpl() 191 const int32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Signed() local 193 tflite::reference_integer_ops::L2Normalization(inputShape.offset, outerSize, axisSize, in l2normQuant8Signed()
|
D | Split.cpp | 33 int outerSize = 1; in splitGeneric() local 35 outerSize *= inputShape.dimensions[i]; in splitGeneric() 44 for (int k = 0; k < outerSize; k++) { in splitGeneric()
|
D | Softmax.cpp | 56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxSlowFloat32() local 60 for (uint32_t outer = 0; outer < outerSize; ++outer) { in softmaxSlowFloat32() 131 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxQuant8Impl() local 135 for (uint32_t outer = 0; outer < outerSize; ++outer) { in softmaxQuant8Impl()
|
D | ArgMinMax.cpp | 31 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl() local 35 for (int outer = 0; outer < outerSize; ++outer) { in argMinMaxImpl()
|
D | TransposeConv2D.cpp | 170 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local 172 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvNhwc() 261 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local 264 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvNhwc() 402 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvQuant8PerChannelNhwc() local 405 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvQuant8PerChannelNhwc()
|
D | LogSoftmax.cpp | 43 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute() local 46 for (uint32_t outer = 0; outer < outerSize; ++outer) { in compute()
|
D | ChannelShuffle.cpp | 40 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local 45 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
|
D | Gather.cpp | 42 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local 47 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
|
D | LocalResponseNormalization.cpp | 56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl() local 60 for (uint32_t outer = 0; outer < outerSize; ++outer) { in localResponseNormFloat32Impl()
|