Lines Matching refs:ExtractSliceOp

250         tensorCast.getOperand().getDefiningOp<ExtractSliceOp>();  in matchAndRewrite()
273 rewriter.replaceOpWithNewOp<ExtractSliceOp>( in matchAndRewrite()
363 if (auto sliceOp = dyn_cast_or_null<tensor::ExtractSliceOp>(definingOp)) { in fold()
963 RankedTensorType ExtractSliceOp::inferResultType( in inferResultType()
976 RankedTensorType ExtractSliceOp::inferResultType( in inferResultType()
987 return ExtractSliceOp::inferResultType(sourceShapedTensorType, staticOffsets, in inferResultType()
999 RankedTensorType ExtractSliceOp::inferCanonicalRankReducedResultType( in inferCanonicalRankReducedResultType()
1023 RankedTensorType ExtractSliceOp::inferCanonicalRankReducedResultType( in inferCanonicalRankReducedResultType()
1035 return ExtractSliceOp::inferCanonicalRankReducedResultType( in inferCanonicalRankReducedResultType()
1042 void ExtractSliceOp::build(OpBuilder &b, OperationState &result, in build()
1060 ExtractSliceOp::inferResultType(sourceRankedTensorType, staticOffsets, in build()
1072 void ExtractSliceOp::build(OpBuilder &b, OperationState &result, Value source, in build()
1082 void ExtractSliceOp::build(OpBuilder &b, OperationState &result, in build()
1096 void ExtractSliceOp::build(OpBuilder &b, OperationState &result, Value source, in build()
1124 LogicalResult ExtractSliceOp::verify() { in verify()
1126 RankedTensorType expectedType = ExtractSliceOp::inferResultType( in verify()
1132 llvm::SmallBitVector ExtractSliceOp::getDroppedDims() { in getDroppedDims()
1152 LogicalResult ExtractSliceOp::reifyResultShapes( in reifyResultShapes()
1188 class ExtractSliceOpCastFolder final : public OpRewritePattern<ExtractSliceOp> {
1190 using OpRewritePattern<ExtractSliceOp>::OpRewritePattern;
1192 LogicalResult matchAndRewrite(ExtractSliceOp sliceOp, in matchAndRewrite()
1209 ExtractSliceOp::inferCanonicalRankReducedResultType( in matchAndRewrite()
1213 Value newSlice = rewriter.create<ExtractSliceOp>( in matchAndRewrite()
1258 : public OpRewritePattern<ExtractSliceOp> {
1260 using OpRewritePattern<ExtractSliceOp>::OpRewritePattern;
1264 : OpRewritePattern<ExtractSliceOp>(context), in ConstantOpExtractSliceFolder()
1267 LogicalResult matchAndRewrite(ExtractSliceOp op, in matchAndRewrite()
1352 RankedTensorType operator()(ExtractSliceOp op, in operator ()()
1356 return ExtractSliceOp::inferCanonicalRankReducedResultType( in operator ()()
1364 void operator()(PatternRewriter &rewriter, ExtractSliceOp op, in operator ()()
1365 ExtractSliceOp newOp) { in operator ()()
1374 void ExtractSliceOp::getCanonicalizationPatterns(RewritePatternSet &results, in getCanonicalizationPatterns()
1378 ExtractSliceOp, SliceReturnTypeCanonicalizer, SliceCanonicalizer>, in getCanonicalizationPatterns()
1406 static Value foldExtractAfterInsertSlice(ExtractSliceOp extractOp) { in foldExtractAfterInsertSlice()
1417 OpFoldResult ExtractSliceOp::fold(ArrayRef<Attribute> operands) { in fold()
1449 return b.createOrFold<tensor::ExtractSliceOp>(loc, targetType, tensor, in createCanonicalRankReducingExtractSliceOp()
1498 RankedTensorType expected = ExtractSliceOp::inferResultType( in verifyInsertSliceOp()
1615 auto sourceType = ExtractSliceOp::inferCanonicalRankReducedResultType( in matchAndRewrite()
2076 auto innerSliceOp = padOp.getSource().getDefiningOp<ExtractSliceOp>(); in matchAndRewrite()
2082 auto outerSliceOp = outerPadOp.getSource().getDefiningOp<ExtractSliceOp>(); in matchAndRewrite()
2180 auto newSliceOp = rewriter.create<ExtractSliceOp>( in matchAndRewrite()
2322 tensor::ExtractSliceOp::inferCanonicalRankReducedResultType( in matchAndRewrite()