/// Print a dense string elements attribute.
void printDenseStringElementsAttr(DenseStringElementsAttr attr);
+ /// Print a dense elements attribute. If 'allowHex' is true, a hex string is
+ /// used instead of individual elements when the elements attr is large.
+ void printDenseIntOrFPElementsAttr(DenseIntOrFPElementsAttr attr,
+ bool allowHex);
+
void printDialectAttribute(Attribute attr);
void printDialectType(Type type);
break;
}
case StandardAttributes::DenseIntOrFPElements: {
- auto eltsAttr = attr.cast<DenseElementsAttr>();
+ auto eltsAttr = attr.cast<DenseIntOrFPElementsAttr>();
if (printerFlags.shouldElideElementsAttr(eltsAttr)) {
printElidedElementsAttr(os);
break;
}
os << "dense<";
- printDenseElementsAttr(eltsAttr, /*allowHex=*/true);
+ printDenseIntOrFPElementsAttr(eltsAttr, /*allowHex=*/true);
os << '>';
break;
}
break;
}
os << "sparse<";
- printDenseElementsAttr(elementsAttr.getIndices(), /*allowHex=*/false);
+ printDenseIntOrFPElementsAttr(elementsAttr.getIndices(),
+ /*allowHex=*/false);
os << ", ";
printDenseElementsAttr(elementsAttr.getValues(), /*allowHex=*/true);
os << '>';
void ModulePrinter::printDenseElementsAttr(DenseElementsAttr attr,
bool allowHex) {
+ if (auto stringAttr = attr.dyn_cast<DenseStringElementsAttr>()) {
+ printDenseStringElementsAttr(stringAttr);
+ return;
+ }
+
+ printDenseIntOrFPElementsAttr(attr.cast<DenseIntOrFPElementsAttr>(),
+ allowHex);
+}
+
+void ModulePrinter::printDenseIntOrFPElementsAttr(DenseIntOrFPElementsAttr attr,
+ bool allowHex) {
auto type = attr.getType();
auto shape = type.getShape();
auto rank = type.getRank();
"foof320"(){bar = sparse<[], []> : tensor<0xf32>} : () -> ()
// CHECK: "foof321"() {bar = sparse<{{\[}}], {{\[}}]> : tensor<f32>} : () -> ()
"foof321"(){bar = sparse<[], []> : tensor<f32>} : () -> ()
+
+// CHECK: "foostr"() {bar = sparse<0, "foo"> : tensor<1x1x1x!unknown<"">>} : () -> ()
+ "foostr"(){bar = sparse<0, "foo"> : tensor<1x1x1x!unknown<"">>} : () -> ()
+// CHECK: "foostr"() {bar = sparse<{{\[\[}}1, 1, 0], {{\[}}0, 1, 0], {{\[}}0, 0, 1]], {{\[}}"a", "b", "c"]> : tensor<2x2x2x!unknown<"">>} : () -> ()
+ "foostr"(){bar = sparse<[[1, 1, 0], [0, 1, 0], [0, 0, 1]], ["a", "b", "c"]> : tensor<2x2x2x!unknown<"">>} : () -> ()
return
}