Searched refs:dimsA (Results 1 – 1 of 1) sorted by relevance
408 inline bool mergeThirdDimension(const T* bufferA, const std::vector<uint32_t>& dimsA, in mergeThirdDimension() argument410 NN_RET_CHECK_EQ(dimsA.size(), 3u); in mergeThirdDimension()413 NN_RET_CHECK_EQ(dimsA[0], dimsB[0]); in mergeThirdDimension()414 NN_RET_CHECK_EQ(dimsA[1], dimsB[1]); in mergeThirdDimension()416 for (unsigned int i = 0; i < dimsA[0]; ++i) { in mergeThirdDimension()417 for (unsigned int j = 0; j < dimsA[1]; ++j) { in mergeThirdDimension()418 for (unsigned int k = 0; k < dimsA[2]; ++k) { in mergeThirdDimension()419 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + k] = in mergeThirdDimension()420 bufferA[(i * dimsA[1] + j) * dimsA[2] + k]; in mergeThirdDimension()423 merged[(i * dimsA[1] + j) * (dimsA[2] + dimsB[2]) + dimsA[2] + k] = in mergeThirdDimension()