Lines Matching defs:extractOp

1090     auto extractOp = operand->get().getDefiningOp<vector::ExtractOp>();
1091     VectorType extractSrcType = extractOp.getSourceVectorType();
1092     Location loc = extractOp.getLoc();
1109           rewriter, warpOp, {extractOp.getVector()},
1110           {extractOp.getSourceVectorType()}, newRetIndices);
1115           loc, distributedVec, extractOp.getMixedPosition());
1140       newDistributedShape[i + extractOp.getNumIndices()] =
1146         rewriter, warpOp, {extractOp.getVector()}, {newDistributedType},
1152         loc, distributedVec, extractOp.getMixedPosition());
1172     auto extractOp = operand->get().getDefiningOp<vector::ExtractOp>();
1173     VectorType extractSrcType = extractOp.getSourceVectorType();
1177           extractOp, "only 0-D or 1-D source supported for now");
1184           extractOp, "only f32/i32 element types are supported");
1200     SmallVector<Value> additionalResults{extractOp.getVector()};
1203         SmallVector<Value>(extractOp.getDynamicPosition()));
1205         SmallVector<Type>(extractOp.getDynamicPosition().getTypes()));
1207     Location loc = extractOp.getLoc();
1227     int64_t staticPos = extractOp.getStaticPosition()[0];
1267     auto extractOp = operand->get().getDefiningOp<vector::ExtractElementOp>();
1269     if (auto pos = extractOp.getPosition()) {
1272     rewriter.setInsertionPoint(extractOp);
1274         extractOp, extractOp.getVector(), indices);