Searched defs:matrixC (Results 1 – 2 of 2) sorted by relevance
| /llvm-project/mlir/lib/Dialect/NVGPU/IR/ |
| H A D | NVGPUDialect.cpp | 127 Value matrixB, Value matrixC, ArrayAttr mmaShape) { in build() argument
134 Value matrixB, Value matrixC, ArrayRef<int64_t> mmaShape, in build() argument
145 TypedValue<VectorType> matrixC, in verifyMmaSyncOp() argument
268 build(::mlir::OpBuilder & odsBuilder,::mlir::OperationState & odsState,Value matrixA,Value matrixB,Value matrixC,Value sparseMetadata,ArrayRef<int64_t> mmaShape) build() argument
556 VectorType matrixC = getMatrixC().getType().getFragmented(); verify() local
[all...] |
| /llvm-project/mlir/lib/Conversion/NVGPUToNVVM/ |
| H A D | NVGPUToNVVM.cpp | 1384 generateWgmma(int i,int j,int k,Value matrixC) generateWgmma() argument
1430 Value matrixC = b.create<LLVM::ExtractValueOp>(adaptor.getMatrixC(), i); generateWgmmaGroup() local
|