Home
last modified time | relevance | path

Searched refs:tensorTy (Results 1 – 4 of 4) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Tensor/Utils/
H A DUtils.cpp62 auto tensorTy = rankedTensor.getType().cast<RankedTensorType>(); in createDynamicDimValues() local
64 for (const auto &en : llvm::enumerate(tensorTy.getShape())) { in createDynamicDimValues()
/llvm-project-15.0.7/mlir/lib/Conversion/ShapeToStandard/
H A DShapeToStandard.cpp552 Type tensorTy = tensor.getType(); in matchAndRewrite() local
553 if (tensorTy.isa<RankedTensorType>()) { in matchAndRewrite()
557 RankedTensorType rankedTensorTy = tensorTy.cast<RankedTensorType>(); in matchAndRewrite()
/llvm-project-15.0.7/mlir/lib/IR/
H A DAsmPrinter.cpp2114 .Case<RankedTensorType>([&](RankedTensorType tensorTy) { in printType() argument
2116 for (int64_t dim : tensorTy.getShape()) { in printType()
2123 printType(tensorTy.getElementType()); in printType()
2125 if (tensorTy.getEncoding()) { in printType()
2127 printAttribute(tensorTy.getEncoding()); in printType()
2131 .Case<UnrankedTensorType>([&](UnrankedTensorType tensorTy) { in printType() argument
2133 printType(tensorTy.getElementType()); in printType()
/llvm-project-15.0.7/flang/lib/Lower/
H A DConvertExpr.cpp7457 mlir::RankedTensorType tensorTy; in createDenseGlobal() local
7462 tensorTy = mlir::RankedTensorType::get(sz, iTy); in createDenseGlobal()
7463 auto init = mlir::DenseElementsAttr::get(tensorTy, ar); in createDenseGlobal()