Searched refs:dimsA (Results 1 – 1 of 1) sorted by relevance
359 inline bool mergeThirdDimension(const T* bufferA, const std::vector<uint32_t>& dimsA, in mergeThirdDimension() argument361 NN_RET_CHECK_EQ(dimsA.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()367 for (unsigned int i = 0; i < dimsA[0]; ++i) { in mergeThirdDimension()368 for (unsigned int j = 0; j < dimsA[1]; ++j) { in mergeThirdDimension()369 for (unsigned int k = 0; k < dimsA[2]; ++k) { in mergeThirdDimension()370 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + k] = in mergeThirdDimension()371 bufferA[(i * dimsA[1] + j) * dimsA[2] + k]; in mergeThirdDimension()374 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + dimsA[2] + k] = in mergeThirdDimension()