Home
last modified time | relevance | path

Searched refs:reifiedReturnShapes (Results 1 – 8 of 8) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/IR/
H A DTensorInferTypeOpInterfaceImpl.cpp153 ReifiedRankedShapedTypeDims &reifiedReturnShapes) const { in reifyResultShapes()
159 reifiedReturnShapes.push_back(getAsValues(b, loc, resultShape)); in reifyResultShapes()
171 ReifiedRankedShapedTypeDims &reifiedReturnShapes) const { in reifyResultShapes()
199 reifiedReturnShapes.emplace_back(std::move(shapes)); in reifyResultShapes()
H A DTensorOps.cpp558 OpBuilder &builder, ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
559 reifiedReturnShapes.resize(1, SmallVector<Value>(getType().getRank())); in reifyResultShapes()
563 reifiedReturnShapes[0][dim] = getOperand(idx++); in reifyResultShapes()
565 reifiedReturnShapes[0][dim] = builder.create<arith::ConstantIndexOp>( in reifyResultShapes()
1153 OpBuilder &builder, ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
1154 reifiedReturnShapes.resize(1); in reifyResultShapes()
1155 reifiedReturnShapes[0].reserve(getType().getRank()); in reifyResultShapes()
1167 reifiedReturnShapes[0].push_back(size.value().get<Value>()); in reifyResultShapes()
1577 OpBuilder &builder, ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
1578 reifiedReturnShapes.resize(1, SmallVector<Value>(getType().getRank())); in reifyResultShapes()
[all …]
/llvm-project-15.0.7/mlir/include/mlir/Interfaces/
H A DInferTypeOpInterface.td177 "::llvm::SmallVectorImpl<::mlir::Value> &":$reifiedReturnShapes),
218 the result shape. The `reifiedReturnShapes` is expected to be
227 "::mlir::ReifiedRankedShapedTypeDims &":$reifiedReturnShapes)
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/IR/
H A DLinalgStructuredOps.td41 ReifiedRankedShapedTypeDims &reifiedReturnShapes) {
43 reifiedReturnShapes);
H A DLinalgInterfaces.td1156 ReifiedRankedShapedTypeDims &reifiedReturnShapes);
/llvm-project-15.0.7/mlir/lib/Dialect/Bufferization/IR/
H A DBufferizationOps.cpp353 OpBuilder &builder, ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
361 reifiedReturnShapes.emplace_back(std::move(shapes)); in reifyResultShapes()
/llvm-project-15.0.7/mlir/lib/Dialect/Linalg/IR/
H A DLinalgInterfaces.cpp564 ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
608 reifiedReturnShapes.emplace_back(std::move(shapes)); in reifyResultShapes()
H A DLinalgOps.cpp1452 OpBuilder &builder, ReifiedRankedShapedTypeDims &reifiedReturnShapes) { in reifyResultShapes() argument
1460 reifiedReturnShapes.emplace_back(std::move(shapes)); in reifyResultShapes()