| /llvm-project/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | BubbleUpExtractSlice.cpp | 121 SmallVector<Type, 4> resultTensorTypes; in matchAndRewrite() local
123 resultTensorTypes.push_back( in matchAndRewrite()
127 clone(rewriter, linalgOp, resultTensorTypes, tiledOperands); in matchAndRewrite()
|
| H A D | Padding.cpp | 209 auto resultTensorTypes = in rewriteAsPaddedOp() local
212 paddedOp = clone(rewriter, opToPad, resultTensorTypes, newOperands); in rewriteAsPaddedOp()
|
| H A D | TilingInterfaceImpl.cpp | 134 SmallVector<Type> resultTensorTypes =
137 Operation *tiledOp = clone(b, linalgOp, resultTensorTypes, tiledOperands);
126 SmallVector<Type> resultTensorTypes = getTiledImplementation() local
|
| H A D | Tiling.cpp | 556 SmallVector<Type> resultTensorTypes = in tileToForallOpImpl()
558 res = clone(b, op, resultTensorTypes, tiledOperands); in tileToForallOpImpl()
740 SmallVector<Type> resultTensorTypes = tileLinalgOpImpl() local
|
| /llvm-project/mlir/include/mlir/Dialect/Linalg/IR/ |
| H A D | LinalgStructuredOps.td | 166 OpBuilder<(ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
172 OpBuilder<(ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
183 OpBuilder<(ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
626 (ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
630 buildMatmulOp($_builder, $_state, resultTensorTypes,
635 (ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
640 buildMatmulOp($_builder, $_state, resultTensorTypes, inputs, outputs,
766 OpBuilder<(ins "TypeRange":$resultTensorTypes, "ValueRange":$inputs,
771 buildStructuredOp($_builder, $_state, resultTensorTypes, inputs,
|
| /llvm-project/mlir/lib/Dialect/Linalg/IR/ |
| H A D | LinalgOps.cpp | 156 /// The result types are derived automatically if `resultTensorTypes` is none. in buildStructuredOp() argument
160 std::optional<TypeRange> resultTensorTypes, in buildStructuredOp()
166 resultTensorTypes.value_or(TypeRange()); in buildStructuredOp()
167 if (!resultTensorTypes) in buildStructuredOp()
188 std::optional<TypeRange> resultTensorTypes, in parseCommonStructuredOpParts()
199 return buildStructuredOp(b, state, resultTensorTypes, inputs, outputs, in parseCommonStructuredOpParts()
999 OpBuilder &builder, OperationState &result, TypeRange resultTensorTypes, in print()
1004 build(builder, result, resultTensorTypes, inputs, outputs, indexingMaps, in print()
1013 OpBuilder &builder, OperationState &result, TypeRange resultTensorTypes, in print()
1019 build(builder, result, resultTensorTypes, input in print()
927 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayAttr indexingMaps,ArrayAttr iteratorTypes,StringAttr doc,StringAttr libraryCall,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
941 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,StringRef doc,StringRef libraryCall,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
982 build(OpBuilder & builder,OperationState & result,TypeRange resultTensorTypes,ValueRange inputs,ValueRange outputs,ArrayRef<AffineMap> indexingMaps,ArrayRef<utils::IteratorType> iteratorTypes,function_ref<void (OpBuilder &,Location,ValueRange)> bodyBuild,ArrayRef<NamedAttribute> attributes) build() argument
[all...] |