Searched defs:OneDimReductionTilingPattern (Results 1 – 1 of 1) sorted by relevance
152 struct OneDimReductionTilingPattern : public OpRewritePattern<GenericOp> { struct153 OneDimReductionTilingPattern(int64_t vector_size, int64_t tile_size, in OneDimReductionTilingPattern() argument160 LogicalResult matchAndRewrite(GenericOp linalg_op, in matchAndRewrite()278 mlir::Optional<Value> getTilableBound(OpBuilder &b, Location loc, Value lb, in getTilableBound()300 SmallVector<Value, 2> TileAndReshapeInputTensors( in TileAndReshapeInputTensors()336 FailureOr<GenericOp> ReduceVectorIntoOutput(PatternRewriter &rewriter, in ReduceVectorIntoOutput()363 int64_t vector_size;364 int64_t tile_size;