Searched refs:dimsB (Results 1 – 1 of 1) sorted by relevance
360 const T* bufferB, const std::vector<uint32_t>& dimsB, T* merged) { in mergeThirdDimension() argument362 NN_RET_CHECK_EQ(dimsB.size(), 3u); in mergeThirdDimension()364 NN_RET_CHECK_EQ(dimsA[0], dimsB[0]); in mergeThirdDimension()365 NN_RET_CHECK_EQ(dimsA[1], dimsB[1]); in mergeThirdDimension()370 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + k] = in mergeThirdDimension()373 for (unsigned int k = 0; k < dimsB[2]; ++k) { in mergeThirdDimension()374 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + dimsA[2] + k] = in mergeThirdDimension()375 bufferB[(i * dimsB[1] + j) * dimsB[2] + k]; in mergeThirdDimension()