Searched refs:inferOperandMMAType (Results 1 – 3 of 3) sorted by relevance
77 Optional<mlir::NVVM::MMATypes> MmaOp::inferOperandMMAType(Type operandElType, in inferOperandMMAType() function in MmaOp98 return inferOperandMMAType(structType.getBody()[0], isAccumulator); in inferOperandMMAType()118 Optional<mlir::NVVM::MMATypes> val = inferOperandMMAType( in accumPtxType()126 inferOperandMMAType(getResult().getType(), /*isAccum=*/true); in resultPtxType()160 inferOperandMMAType(regTypes.back(), /*isAccum=*/fragIdx >= 2); in print()211 if (auto res = inferOperandMMAType(operandA[0].getType(), false)) in build()213 if (auto res = inferOperandMMAType(operandB[0].getType(), false)) in build()301 inferOperandMMAType(frag.regTypes[0], /*isAccum=*/iter.index() < 2); in parse()307 frags[3].elemtype = inferOperandMMAType(resultType, /*isAccum=*/true); in parse()
276 Optional<NVVM::MMATypes> ptxTypeC = NVVM::MmaOp::inferOperandMMAType( in matchAndRewrite()
976 static Optional<mlir::NVVM::MMATypes> inferOperandMMAType(Type operandElType,