Searched refs:indicesType (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/AsmParser/ |
| H A D | AttributeParser.cpp | 1006 ShapedType indicesType = in parseSparseElementsAttr() local
1010 loc, type, DenseElementsAttr::get(indicesType, ArrayRef<Attribute>()), in parseSparseElementsAttr()
1042 ShapedType indicesType; in parseSparseElementsAttr() local
1044 indicesType = RankedTensorType::get({1, type.getRank()}, indiceEltType); in parseSparseElementsAttr()
1047 indicesType = RankedTensorType::get(indiceParser.getShape(), indiceEltType); in parseSparseElementsAttr()
1049 auto indices = indiceParser.getAttr(indicesLoc, indicesType); in parseSparseElementsAttr()
1057 ? RankedTensorType::get({indicesType.getDimSize(0)}, valuesEltType) in parseSparseElementsAttr()
|
| /llvm-project-15.0.7/mlir/unittests/Dialect/Quant/ |
| H A D | QuantizationUtilsTest.cpp | 62 auto indicesType = RankedTensorType::get({1, 2}, IntegerType::get(ctx, 64)); in getTestSparseElementsAttr() local
64 DenseIntElementsAttr::get(indicesType, {APInt(64, 0), APInt(64, 0)}); in getTestSparseElementsAttr()
|
| /llvm-project-15.0.7/mlir/unittests/IR/ |
| H A D | AttributeTest.cpp | 219 auto indicesType = in TEST() local
222 DenseIntElementsAttr::get(indicesType, {APInt(64, 0), APInt(64, 0)}); in TEST()
|
| /llvm-project-15.0.7/mlir/lib/IR/ |
| H A D | BuiltinAttributes.cpp | 1672 ShapedType indicesType = sparseIndices.getType(); in verify() local
1676 << indicesType.getShape() in verify()
1681 size_t rank = type.getRank(), indicesRank = indicesType.getRank(); in verify()
1683 if (indicesType.getDimSize(1) != static_cast<int64_t>(rank)) in verify()
1689 int64_t numSparseIndices = indicesType.getDimSize(0); in verify()
|