Searched defs:outerSize (Results 1 – 9 of 9) sorted by relevance
/aosp12/packages/modules/NeuralNetworks/common/operations/ |
H A D | L2Normalization.cpp | 52 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 D | ArgMinMax.cpp | 31 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl() local
|
H A D | Gather.cpp | 42 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
|
H A D | LogSoftmax.cpp | 43 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute() local
|
H A D | ChannelShuffle.cpp | 40 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
|
H A D | Softmax.cpp | 56 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 D | Split.cpp | 33 int outerSize = 1; in splitGeneric() local
|
H A D | LocalResponseNormalization.cpp | 56 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl() local
|
H A D | TransposeConv2D.cpp | 170 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
|