Searched defs:contractOp (Results 1 – 5 of 5) sorted by relevance
52 if (auto contractOp = dyn_cast<vector::ContractionOp>(user)) in getUserContract() local 75 FailureOr<vector::ContractionOp> contractOp = getUserContract(op); in getWarpMatrixInfo() local
186 SmallVector<int64_t> nativeShape(contractOp.getIteratorTypes().size(), in runOnOperation() local 214 if (!contractOp) in runOnOperation() local 202 vector::ContractionOp contractOp = runOnOperation() local
333 mlir::vector::castAwayContractionLeadingOneDim(vector::ContractionOp contractOp, in castAwayContractionLeadingOneDim() argument [all...]
269 auto contractOp = resTOp.getVector().getDefiningOp<vector::ContractionOp>(); in matchAndRewrite() local [all...]
258 FailureOr<vector::ContractionOp> contractOp = nvgpu::getUserContract(op); extractStridedSliceSupportsMMAMatrixType() local 1254 } else if (auto contractOp = dyn_cast<vector::ContractionOp>(op)) { convertVectorToMMAOps() local