Home
last modified time | relevance | path

Searched defs:innerSize (Results 1 – 7 of 7) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DL2Normalization.cpp54 const uint32_t innerSize = in l2normFloat32Impl() local
81 const uint32_t innerSize = in l2normQuant8Impl() local
113 const uint32_t innerSize = in l2normQuant8SignedImpl() local
H A DArgMinMax.cpp33 const int innerSize = in argMinMaxImpl() local
H A DGather.cpp44 const auto innerSize = in eval() local
H A DLogSoftmax.cpp45 const uint32_t innerSize = getNumberOfElements(shape, axis + 1, getNumberOfDimensions(shape)); in compute() local
H A DChannelShuffle.cpp42 const uint32_t innerSize = in eval() local
H A DSoftmax.cpp58 const uint32_t innerSize = in softmaxSlowFloat32() local
133 const uint32_t innerSize = in softmaxQuant8Impl() local
H A DLocalResponseNormalization.cpp58 const uint32_t innerSize = in localResponseNormFloat32Impl() local