Searched refs:secondDimSize (Results 1 – 2 of 2) sorted by relevance
52 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local55 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()57 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()
69 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local72 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()74 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()