Searched refs:shapedOp (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| D | Tiling.cpp | 250 Value shapedOp = en.value(); in makeTiledShapes() local
251 ShapedType shapedType = shapedOp.getType().cast<ShapedType>(); in makeTiledShapes()
256 res.push_back(shapedOp); in makeTiledShapes()
268 sizes.push_back(std_dim(shapedOp, r)); in makeTiledShapes()
296 auto d = std_dim(shapedOp, r); in makeTiledShapes()
307 b.create<SubViewOp>(loc, shapedOp, offsets, sizes, strides)); in makeTiledShapes()
310 b.create<SubTensorOp>(loc, shapedOp, offsets, sizes, strides)); in makeTiledShapes()
|
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestPatterns.cpp | 128 auto shapedOp = cast<OpWithShapedTypeInferTypeInterfaceOp>(op); in reifyReturnShape() local
130 if (failed(shapedOp.reifyReturnTypeShapes(b, shapes))) in reifyReturnShape()
|