Searched refs:numDimensions (Results 1 – 4 of 4) sorted by relevance
186 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare() local189 NN_RET_CHECK_LT(axis, numDimensions); in prepare()190 NN_RET_CHECK_LE(numDimensions, 4); in prepare()195 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()197 for (uint32_t d = 0; d < numDimensions; ++d) { in prepare()
185 int32_t numDimensions = getNumberOfDimensions(input); in prepare() local189 NN_RET_CHECK_LE(numDimensions, 4); in prepare()190 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()191 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
239 int32_t numDimensions = getNumberOfDimensions(input); in prepare() local243 NN_RET_CHECK_LE(numDimensions, 4); in prepare()244 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()245 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
1085 int numDimensions = 0;1087 numDimensions++;1091 if (numDimensions > 0) {1110 for (int i = 0; i < numDimensions; ++i) {