Lines Matching defs:matmul

76 // Return true if the contract op can be convert to MMA matmul.
91   // The contract needs to represent a matmul to be able to convert to
92   // MMAMatrix matmul.
385 // to MMA matmul.
412     // This is the classical row-major matmul, nothing to do.
515 // MMA types have different layout based on how they are used in matmul ops.
1040   Value matmul = rewriter.create<gpu::SubgroupMmaComputeOp>(
1043   valueMapping[op.getResult()] = matmul;
1063   Value matmul = rewriter.create<nvgpu::MmaSyncOp>(
1065   valueMapping[op.getResult()] = matmul;