Searched refs:operandElType (Results 1 – 2 of 2) sorted by relevance
77 Optional<mlir::NVVM::MMATypes> MmaOp::inferOperandMMAType(Type operandElType, in inferOperandMMAType() argument80 LLVM::getFixedVectorType(Float16Type::get(operandElType.getContext()), 2); in inferOperandMMAType()81 if (operandElType.isF64()) in inferOperandMMAType()83 if (operandElType.isF16() || operandElType == half2Type) in inferOperandMMAType()85 if (operandElType.isF32() && isAccumulator) in inferOperandMMAType()87 if (operandElType.isF32() && !isAccumulator) in inferOperandMMAType()89 if (operandElType.isa<IntegerType>()) { in inferOperandMMAType()95 if (auto structType = operandElType.dyn_cast<LLVM::LLVMStructType>()) { in inferOperandMMAType()
976 static Optional<mlir::NVVM::MMATypes> inferOperandMMAType(Type operandElType,