Searched refs:ndim (Results 1 – 3 of 3) sorted by relevance
76 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxFloat32() local79 if (axis == ndim - 1) { in softmaxFloat32()188 int32_t ndim = getNumberOfDimensions(inputShape); in softmaxQuant8() local209 if (axis == ndim - 1) { in softmaxQuant8()245 const size_t ndim = context->getInputShape(kInputTensor).dimensions.size(); in validate() local246 if (ndim != 2 && ndim != 4 && ndim != 0) { in validate()
100 int32_t ndim = getNumberOfDimensions(inputShape); in l2normFloat32() local103 if (axis == ndim - 1) { in l2normFloat32()129 int32_t ndim = getNumberOfDimensions(inputShape); in l2normQuant8() local132 if (axis == ndim - 1) { in l2normQuant8()
77 int32_t ndim = getNumberOfDimensions(inputShape); in localResponseNormFloat32() local80 if (axis == ndim - 1) { in localResponseNormFloat32()