Searched defs:mmaShape (Results 1 – 3 of 3) 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
146 const std::array<int64_t, 3> &mmaShape, in verifyMmaSyncOp() argument
269 build(::mlir::OpBuilder & odsBuilder,::mlir::OperationState & odsState,Value matrixA,Value matrixB,Value matrixC,Value sparseMetadata,ArrayRef<int64_t> mmaShape) build() argument
|
| /llvm-project/mlir/lib/Dialect/LLVMIR/IR/ |
| H A D | NVVMDialect.cpp | 393 std::array<int64_t, 3> mmaShape{getShapeAttr().getM(), getShapeAttr().getN(), verify() local
|
| /llvm-project/mlir/lib/Dialect/NVGPU/TransformOps/ |
| H A D | NVGPUTransformOps.cpp | 462 SmallVector<int64_t> mmaShape; member
|