Searched refs:elementsPerRegister (Results 1 – 3 of 3) sorted by relevance
158 int64_t elementsPerRegister, in getRegisterIndexToTileOffsetMap() argument164 AffineExpr registerIdx = logicalValueId.floorDiv(elementsPerRegister); in getRegisterIndexToTileOffsetMap()183 const int64_t elementsPerRegister = in getLaneIdAndValueIdToOperandCoord() local194 isAccumulatorOrResult(fragmentType.operandRole), elementsPerRegister, in getLaneIdAndValueIdToOperandCoord()204 tileCol + (laneId % kThreadsPerRow) * elementsPerRegister + in getLaneIdAndValueIdToOperandCoord()205 (logicalValueIdDim % elementsPerRegister)}); in getLaneIdAndValueIdToOperandCoord()
49 int64_t elementsPerRegister; member
435 regInfo.elementsPerRegister}; in getMmaSyncVectorOperandType()556 builder.getIndexAttr(i * regInfo->elementsPerRegister)); in createNonLdMatrixLoads()576 builder.getIndexAttr(i * regInfo->elementsPerRegister + innerIdx)); in createNonLdMatrixLoads()653 b.getIndexAttr(i * regInfo->elementsPerRegister)); in convertTransferWriteToStores()