Home
last modified time | relevance | path

Searched refs:ndim (Results 1 – 4 of 4) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DL2Normalization.cpp139 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32() local
142 if (axis == ndim - 1) { in l2normFloat32()
168 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8() local
171 if (axis == ndim - 1) { in l2normQuant8()
184 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8Signed() local
187 if (axis == ndim - 1) { in l2normQuant8Signed()
DSoftmax.cpp84 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32() local
87 if (axis == ndim - 1) { in softmaxFloat32()
202 [[maybe_unused]] int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8() local
DLocalResponseNormalization.cpp83 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNorm() local
87 if (axis == ndim - 1) { in localResponseNorm()
/packages/modules/NeuralNetworks/runtime/include/
DNeuralNetworksWrapper.h172 void updateDimensions(std::vector<uint32_t> ndim) { in updateDimensions()
173 dimensions = ndim; in updateDimensions()