Searched refs:dimsB (Results 1 – 1 of 1) sorted by relevance
376 const T* bufferB, const std::vector<uint32_t>& dimsB, T* merged) { in mergeThirdDimension() argument378 NN_RET_CHECK_EQ(dimsB.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()386 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + k] = in mergeThirdDimension()389 for (unsigned int k = 0; k < dimsB[2]; ++k) { in mergeThirdDimension()390 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + dimsA[2] + k] = in mergeThirdDimension()391 bufferB[(i * dimsB[1] + j) * dimsB[2] + k]; in mergeThirdDimension()