Lines Matching defs:spMatA
607 Value spMatA = spGenA->getResult(0);
621 loc, indexTp, tokenTp, token, spMatA, dnX, dnY,
631 loc, tokenTp, token, spMatA, dnX, dnY, /*computeType=*/dnYType, buffer);
635 token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
705 Value spMatA = spGenA->getResult(0);
721 loc, indexTp, tokenTp, token, spMatA, dnB, dnC,
732 loc, tokenTp, token, spMatA, dnB, dnC, /*computeType=*/dnCType, buffer);
736 token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
810 Value spMatA = spGenA->getResult(0);
844 gpu::TransposeMode::NON_TRANSPOSE, spMatA, spMatB, spMatC, dnCType, zero,
853 gpu::TransposeMode::NON_TRANSPOSE, spMatA, spMatB, spMatC, dnCType,
861 gpu::TransposeMode::NON_TRANSPOSE, spMatA, spMatB, spMatC, dnCType, zero,
870 gpu::TransposeMode::NON_TRANSPOSE, spMatA, spMatB, spMatC, dnCType,
892 gpu::TransposeMode::NON_TRANSPOSE, spMatA, spMatB, spMatC, dnCType);
903 token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)
983 Value spMatA = spGenA->getResult(0);
1002 gpu::TransposeMode::NON_TRANSPOSE, spMatA, dnB, dnC,
1023 loc, tokenTp, token, spMatA, dnB, dnC, /*computeType=*/dnCType,
1028 token = rewriter.create<gpu::DestroySpMatOp>(loc, tokenTp, token, spMatA)