Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/
DOperationsUtils.cpp352 uint32_t num_dimensions = getNumberOfDimensions(inputShapes[0]); in concatenationPrepare() local
355 NN_OPS_CHECK(axis < (int32_t)num_dimensions); in concatenationPrepare()
359 NN_OPS_CHECK(getNumberOfDimensions(inputShapes[i]) == num_dimensions); in concatenationPrepare()
365 for (int d = 0; d < (int32_t)num_dimensions; ++d) { in concatenationPrepare()