Searched refs:dimsA (Results 1 – 1 of 1) sorted by relevance
375 inline bool mergeThirdDimension(const T* bufferA, const std::vector<uint32_t>& dimsA, in mergeThirdDimension() argument377 NN_RET_CHECK_EQ(dimsA.size(), 3u); in mergeThirdDimension()380 NN_RET_CHECK_EQ(dimsA[0], dimsB[0]); in mergeThirdDimension()381 NN_RET_CHECK_EQ(dimsA[1], dimsB[1]); in mergeThirdDimension()383 for (unsigned int i = 0; i < dimsA[0]; ++i) { in mergeThirdDimension()384 for (unsigned int j = 0; j < dimsA[1]; ++j) { in mergeThirdDimension()385 for (unsigned int k = 0; k < dimsA[2]; ++k) { in mergeThirdDimension()386 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + k] = in mergeThirdDimension()387 bufferA[(i * dimsA[1] + j) * dimsA[2] + k]; in mergeThirdDimension()390 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + dimsA[2] + k] = in mergeThirdDimension()