Searched refs:num_dimensions (Results 1 – 2 of 2) sorted by relevance
145 inline int ReverseMaskBits(int mask, int num_dimensions) { in ReverseMaskBits() argument147 for (int dim = 0; dim < num_dimensions; dim++) { in ReverseMaskBits()
377 uint32_t num_dimensions = getNumberOfDimensions(inputShapes[0]); in concatenationPrepare() local380 NN_OPS_CHECK(axis < (int32_t)num_dimensions); in concatenationPrepare()384 NN_OPS_CHECK(getNumberOfDimensions(inputShapes[i]) == num_dimensions); in concatenationPrepare()390 for (int d = 0; d < (int32_t)num_dimensions; ++d) { in concatenationPrepare()