Lines Matching refs:matrixA

126                       ::mlir::OperationState &odsState, Value matrixA,
128   build(odsBuilder, odsState, matrixC.getType(), matrixA, matrixB, matrixC,
133                       ::mlir::OperationState &odsState, Value matrixA,
136   build(odsBuilder, odsState, matrixC.getType(), matrixA, matrixB, matrixC,
143                                      TypedValue<VectorType> matrixA,
166   auto aVector = matrixA.getType();
207     return op->emitError() << "matrixA must be 2 dimensional vector";
279                             ::mlir::OperationState &odsState, Value matrixA,
282   build(odsBuilder, odsState, matrixC.getType(), matrixA, matrixB, matrixC,
566   MemRefType matrixA = getDescriptorA().getType().getTensor();
574   if (matrixA.getRank() != 2 || matrixB.getRank() != 2 ||
580   if (matrixA.getShape()[1] != matrixB.getShape()[0])
581     return emitOpError() << "2nd dim matrix-A (" << matrixA.getShape()[1]
584   if (matrixA.getShape()[0] != matrixC.getShape()[0])
585     return emitOpError() << "1st dim matrix-A ( " << matrixA.getShape()[0]
594                                     matrixA.getElementType(),
597                          << " += " << matrixA.getElementType() << " * "
601   if (failed(isAllowedSizeN(matrixB.getDimSize(1), matrixA.getElementType()))) {
607   if (!matrixC.getElementType().isF32() && !matrixA.getElementType().isF16() &&
608       !matrixA.getElementType().isBF16()) {
610                          << " += " << matrixA.getElementType() << " * "