Searched refs:outerSize (Results 1 – 9 of 9) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 32 int outerSize = 1; in splitGeneric() local
34 outerSize *= inputShape.dimensions[i]; in splitGeneric()
43 for (int k = 0; k < outerSize; k++) { in splitGeneric()
|
| D | ArgMinMax.cpp | 33 const int outerSize = getNumberOfElements(inputShape, 0, axis); in argMinMaxImpl() local
37 for (int outer = 0; outer < outerSize; ++outer) { in argMinMaxImpl()
|
| D | L2Normalization.cpp | 42 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normFloat32Impl() local
46 for (uint32_t outer = 0; outer < outerSize; ++outer) { in l2normFloat32Impl()
69 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in l2normQuant8Impl() local
73 for (uint32_t outer = 0; outer < outerSize; ++outer) { in l2normQuant8Impl()
|
| D | Normalization.cpp | 34 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in localResponseNormFloat32Impl() local
38 for (uint32_t outer = 0; outer < outerSize; ++outer) { in localResponseNormFloat32Impl()
|
| D | Softmax.cpp | 45 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxSlowFloat32() local
49 for (uint32_t outer = 0; outer < outerSize; ++outer) { in softmaxSlowFloat32()
119 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in softmaxQuant8Impl() local
123 for (uint32_t outer = 0; outer < outerSize; ++outer) { in softmaxQuant8Impl()
|
| D | ChannelShuffle.cpp | 41 const uint32_t outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
46 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
|
| D | LogSoftmax.cpp | 42 const uint32_t outerSize = getNumberOfElements(shape, 0, axis); in compute() local
45 for (uint32_t outer = 0; outer < outerSize; ++outer) { in compute()
|
| D | Gather.cpp | 43 const auto outerSize = getNumberOfElements(inputShape, 0, axis); in eval() local
48 for (uint32_t outer = 0; outer < outerSize; ++outer) { in eval()
|
| D | TransposeConv2D.cpp | 159 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local
161 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvNhwc()
250 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvNhwc() local
253 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvNhwc()
390 const uint32_t outerSize = numBatches * outputHeight * outputWidth; in transposeConvQuant8PerChannelNhwc() local
393 for (uint32_t i = 0; i < outerSize; i++) { in transposeConvQuant8PerChannelNhwc()
|