Searched refs:firstDimSize (Results 1 – 2 of 2) sorted by relevance
42 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims() local45 for (int f = 0; f < firstDimSize; ++f) { in transposeFirstTwoDims()49 const uint32_t outputIndex = s * firstDimSize * inputSize + f * inputSize + i; in transposeFirstTwoDims()
54 const uint32_t firstDimSize = getSizeOfDimension(inputShape, 0); in transposeFirstTwoDims() local57 for (int f = 0; f < firstDimSize; ++f) { in transposeFirstTwoDims()61 const uint32_t outputIndex = s * firstDimSize * inputSize + f * inputSize + i; in transposeFirstTwoDims()