Searched refs:firstDim (Results 1 – 2 of 2) sorted by relevance
345 const int firstDim = getSizeOfDimension(shape, 0); in transposeFirstTwoDimensions() local352 for (int i = 0; i < firstDim; ++i) { in transposeFirstTwoDimensions()355 transposedBuffer[(j * firstDim + i) * blockSize + k] = in transposeFirstTwoDimensions()
279 size_t firstDim = (first == DimensionKind::UNKNOWN_AT_COMPILE_OTHER_AT_EXECUTE) in TestOne() local285 return firstDim * secondDim * sizeof(fives[0][0]); in TestOne()