Searched refs:contractionOp (Results 1 – 1 of 1) sorted by relevance
458 vector::ContractionOp contractionOp, ArrayRef<int64_t> targetShape, in getVectorContractionOpUnrollState() argument462 contractionOp.getIterationIndexMap(iterationIndexMapList); in getVectorContractionOpUnrollState()467 vectors[i].type = contractionOp.getOperand(i).getType().cast<VectorType>(); in getVectorContractionOpUnrollState()473 if (llvm::size(contractionOp.masks()) == 2) { in getVectorContractionOpUnrollState()476 vectors.push_back({contractionOp.getLHSVectorMaskType(), in getVectorContractionOpUnrollState()478 vectors.push_back({contractionOp.getRHSVectorMaskType(), in getVectorContractionOpUnrollState()682 if (auto contractionOp = dyn_cast<vector::ContractionOp>(op)) { in unrollSingleResultVectorOp() local684 getVectorContractionOpUnrollState(contractionOp, targetShape, vectors, in unrollSingleResultVectorOp()