Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DL2Normalization.cpp52 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl() local
79 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl() local
111 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8SignedImpl() local
191 const int32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Signed() local
H A DArgMinMax.cpp31 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl() local
H A DGather.cpp42 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
H A DLogSoftmax.cpp43 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute() local
H A DChannelShuffle.cpp40 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
H A DSoftmax.cpp56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxSlowFloat32() local
131 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxQuant8Impl() local
H A DSplit.cpp33 int outerSize = 1; in splitGeneric() local
H A DLocalResponseNormalization.cpp56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl() local
H A DTransposeConv2D.cpp170 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local
261 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local
402 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvQuant8PerChannelNhwc() local