Searched refs:secondDimSize (Results 1 – 2 of 2) sorted by relevance
51 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local54 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()56 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()
67 const uint32_t secondDimSize = getSizeOfDimension(inputShape, 1); in transposeFirstTwoDims() local70 for (int s = 0; s < secondDimSize; ++s) { in transposeFirstTwoDims()72 const uint32_t inputIndex = f * secondDimSize * inputSize + s * inputSize + i; in transposeFirstTwoDims()