Home
last modified time | relevance | path

Searched defs:matrixB (Results 1 – 3 of 3) sorted by relevance

/llvm-project/mlir/lib/Dialect/NVGPU/IR/
H A DNVGPUDialect.cpp127 Value matrixB, Value matrixC, ArrayAttr mmaShape) { in build() argument
134 Value matrixB, Value matrixC, ArrayRef<int64_t> mmaShape, in build() argument
144 TypedValue<VectorType> matrixB, 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
555 MemRefType matrixB = getDescriptorB().getType().getTensor(); verify() local
[all...]
/llvm-project/flang/unittests/Optimizer/Builder/Runtime/
H A DTransformationalTest.cpp160 mlir::Value matrixB = builder.create<fir::UndefOp>(loc, boxTy2); in testGenMatmul() local
/llvm-project/flang/lib/Optimizer/Builder/
H A DIntrinsicCall.cpp5326 mlir::Value matrixB = fir::getBase(matrixTmpB); genMatmul() local
5353 mlir::Value matrixB = fir::getBase(matrixTmpB); genMatmulTranspose() local