Home
last modified time | relevance | path

Searched refs:numDimensions (Results 1 – 3 of 3) sorted by relevance

/frameworks/ml/nn/common/operations/
DConcatenation.cpp180 uint32_t numDimensions = getNumberOfDimensions(input0); in prepare() local
183 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
184 NN_RET_CHECK_LE(numDimensions, 4); in prepare()
189 NN_RET_CHECK_EQ(getNumberOfDimensions(input), numDimensions); in prepare()
191 for (uint32_t d = 0; d < numDimensions; ++d) { in prepare()
DLocalResponseNormalization.cpp179 int32_t numDimensions = getNumberOfDimensions(input); in prepare() local
183 NN_RET_CHECK_LE(numDimensions, 4); in prepare()
184 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()
185 NN_RET_CHECK_LT(axis, numDimensions); in prepare()
DL2Normalization.cpp234 int32_t numDimensions = getNumberOfDimensions(input); in prepare() local
238 NN_RET_CHECK_LE(numDimensions, 4); in prepare()
239 NN_RET_CHECK_GE(axis, -numDimensions); in prepare()
240 NN_RET_CHECK_LT(axis, numDimensions); in prepare()