Searched refs:secondDimSize (Results 1 – 2 of 2) sorted by relevance
43 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local46 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()48 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()
55 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local58 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()60 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()