Home
last modified time | relevance | path

Searched refs:outerSize (Results 1 – 9 of 9) sorted by relevance

/packages/modules/NeuralNetworks/common/operations/
DL2Normalization.cpp52 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()
DSplit.cpp33 int outerSize = 1; in splitGeneric() local
35 outerSize *= inputShape.dimensions[i]; in splitGeneric()
44 for (int k = 0; k < outerSize; k++) { in splitGeneric()
DSoftmax.cpp56 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()
DArgMinMax.cpp31 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl() local
35 for (int outer = 0; outer < outerSize; ++outer) { in argMinMaxImpl()
DTransposeConv2D.cpp170 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()
DLogSoftmax.cpp43 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute() local
46 for (uint32_t outer = 0; outer < outerSize; ++outer) { in compute()
DChannelShuffle.cpp40 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
45 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
DGather.cpp42 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
47 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
DLocalResponseNormalization.cpp56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl() local
60 for (uint32_t outer = 0; outer < outerSize; ++outer) { in localResponseNormFloat32Impl()