Searched refs:indicesType (Results 1 – 2 of 2) sorted by relevance
| /external/llvm-project/mlir/lib/Parser/ |
| D | AttributeParser.cpp | 925 ShapedType indicesType = in parseSparseElementsAttr() local
929 type, DenseElementsAttr::get(indicesType, ArrayRef<Attribute>()), in parseSparseElementsAttr()
961 ShapedType indicesType; in parseSparseElementsAttr() local
963 indicesType = RankedTensorType::get({1, type.getRank()}, indiceEltType); in parseSparseElementsAttr()
966 indicesType = RankedTensorType::get(indiceParser.getShape(), indiceEltType); in parseSparseElementsAttr()
968 auto indices = indiceParser.getAttr(indicesLoc, indicesType); in parseSparseElementsAttr()
976 ? RankedTensorType::get({indicesType.getDimSize(0)}, valuesEltType) in parseSparseElementsAttr()
984 auto sameShape = (indicesType.getRank() == 1) || in parseSparseElementsAttr()
985 (type.getRank() == indicesType.getDimSize(1)); in parseSparseElementsAttr()
986 auto sameElementNum = indicesType.getDimSize(0) == valuesType.getDimSize(0); in parseSparseElementsAttr()
[all …]
|
| /external/llvm-project/mlir/unittests/Dialect/Quant/ |
| D | QuantizationUtilsTest.cpp | 62 auto indicesType = RankedTensorType::get({1, 2}, IntegerType::get(64, ctx)); in getTestSparseElementsAttr() local
64 DenseIntElementsAttr::get(indicesType, {APInt(64, 0), APInt(64, 0)}); in getTestSparseElementsAttr()
|