Searched refs:inferResultType (Results 1 – 9 of 9) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/ |
| H A D | TensorOps.cpp | 963 RankedTensorType ExtractSliceOp::inferResultType( in inferResultType() function in ExtractSliceOp
976 RankedTensorType ExtractSliceOp::inferResultType( in inferResultType() function in ExtractSliceOp
1005 inferResultType(sourceRankedTensorType, offsets, sizes, strides) in inferCanonicalRankReducedResultType()
1060 ExtractSliceOp::inferResultType(sourceRankedTensorType, staticOffsets, in build()
1126 RankedTensorType expectedType = ExtractSliceOp::inferResultType( in verify()
1498 RankedTensorType expected = ExtractSliceOp::inferResultType( in verifyInsertSliceOp()
1826 auto expectedType = PadOp::inferResultType( in verify()
1865 RankedTensorType PadOp::inferResultType(RankedTensorType sourceType, in inferResultType() function in PadOp
1899 auto resultType = inferResultType(sourceType, staticLow, staticHigh); in build()
1932 resultType = PadOp::inferResultType(sourceType, staticLow, staticHigh); in build()
[all …]
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgOps.td | 62 static Type inferResultType(ArrayRef<int64_t> staticSizes, Type elementType,
108 InitTensorOp::inferResultType(staticShape, elementType),
|
| /llvm-project-15.0.7/mlir/lib/Dialect/MemRef/IR/ |
| H A D | MemRefOps.cpp | 2086 Type SubViewOp::inferResultType(MemRefType sourceMemRefType, in inferResultType() function in SubViewOp
2135 Type SubViewOp::inferResultType(MemRefType sourceMemRefType, in inferResultType() function in SubViewOp
2147 return SubViewOp::inferResultType(sourceMemRefType, staticOffsets, in inferResultType()
2157 inferResultType(sourceRankedTensorType, offsets, sizes, strides) in inferRankReducedResultType()
2216 resultType = SubViewOp::inferResultType(sourceMemRefType, staticOffsets, in build()
2386 auto expectedType = SubViewOp::inferResultType( in verify()
2441 auto nonRankReducedType = SubViewOp::inferResultType(sourceType, mixedOffsets, in getCanonicalSubViewResultType()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | Fusion.cpp | 162 resultTypes.push_back(tensor::ExtractSliceOp::inferResultType( in fuse()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Tensor/IR/ |
| H A D | TensorOps.td | 308 static RankedTensorType inferResultType(
313 static RankedTensorType inferResultType(
990 static RankedTensorType inferResultType(
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 1215 Type expectedType = InitTensorOp::inferResultType( in verify()
1225 Type InitTensorOp::inferResultType(ArrayRef<int64_t> staticSizes, in inferResultType() function in InitTensorOp
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/MemRef/IR/ |
| H A D | MemRefOps.td | 1640 static Type inferResultType(MemRefType sourceMemRefType,
1644 static Type inferResultType(MemRefType sourceMemRefType,
|
| /llvm-project-15.0.7/mlir/lib/Conversion/MemRefToLLVM/ |
| H A D | MemRefToLLVM.cpp | 1652 memref::SubViewOp::inferResultType( in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 3254 tensor::ExtractSliceOp::inferResultType( in matchAndRewrite()
3756 tensor::ExtractSliceOp::inferResultType( in matchAndRewrite()
|