| /llvm-project-15.0.7/mlir/lib/Interfaces/ |
| H A D | InferTypeOpInterface.cpp | 45 void ShapeAdaptor::getDims(SmallVectorImpl<int64_t> &res) const { in getDims() function in ShapeAdaptor
57 auto vals = val.get<ShapedTypeComponents *>()->getDims(); in getDims()
62 void ShapeAdaptor::getDims(ShapedTypeComponents &res) const { in getDims() function in ShapeAdaptor
65 getDims(res.dims); in getDims()
77 return stc->getDims()[index]; in getDimSize()
86 return val.get<ShapedTypeComponents *>()->getDims().size(); in getRank()
103 return llvm::none_of(stc->getDims(), ShapedType::isDynamic); in hasStaticShape()
124 for (int64_t dim : stc->getDims()) { in getNumElements()
138 getDims(dims); in dump()
195 shapeAndType.getDims(), shapeAndType.getElementType())); in inferReturnTensorTypes()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | InferTypeOpInterface.h | 54 void getDims(SmallVectorImpl<int64_t> &res) const;
58 void getDims(ShapedTypeComponents &res) const;
120 adaptor.getDims(*this); in ShapedTypeComponents()
135 ArrayRef<int64_t> getDims() const { in getDims() function
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaInferShapes.cpp | 53 Type newType = oldType.clone(inferredTy.getDims()); in propagateShapesToTosaIf()
88 auto newTy = operandTy.clone(shapedTypeComponent.getDims()); in propagateShapesToTosaWhile()
247 for (auto dim : predictedShape.getDims()) { in propagateShapesInRegion()
270 Type t = it.first.getType().cast<ShapedType>().clone(result.getDims()); in propagateShapesInRegion()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/ |
| H A D | MLIRGen.cpp | 253 auto type = getType(lit.getDims()); in mlirGen()
258 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in mlirGen()
265 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/ |
| H A D | MLIRGen.cpp | 253 auto type = getType(lit.getDims()); in mlirGen()
258 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in mlirGen()
265 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/ |
| H A D | MLIRGen.cpp | 257 auto type = getType(lit.getDims()); in mlirGen()
262 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in mlirGen()
269 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/ |
| H A D | MLIRGen.cpp | 257 auto type = getType(lit.getDims()); in mlirGen()
262 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in mlirGen()
269 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/ |
| H A D | MLIRGen.cpp | 257 auto type = getType(lit.getDims()); in mlirGen()
262 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in mlirGen()
269 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/ |
| H A D | MLIRGen.cpp | 393 data.reserve(std::accumulate(lit.getDims().begin(), lit.getDims().end(), 1, in getConstantAttr()
400 auto dataType = mlir::RankedTensorType::get(lit.getDims(), elementType); in getConstantAttr()
447 mlir::Type type = getType(lit.getDims()); in mlirGen()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch1/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch2/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| H A D | AST.h | 89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims() function
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch3/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch4/include/toy/ |
| H A D | Parser.h | 139 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
148 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| H A D | AST.h | 89 llvm::ArrayRef<int64_t> getDims() { return dims; } in getDims() function
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch3/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch1/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch2/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch4/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/parser/ |
| H A D | AST.cpp | 130 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/include/toy/ |
| H A D | Parser.h | 154 auto firstDims = firstLiteral->getDims(); in parseTensorLiteralExpr()
163 if (exprLiteral->getDims() != firstDims) in parseTensorLiteralExpr()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/parser/ |
| H A D | AST.cpp | 133 llvm::interleaveComma(literal->getDims(), llvm::errs()); in printLitHelper()
|