We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
XLA_EXPERIMENTAL=nonzero:masked_select:nms python3 test/ds/test_dynamic_shapes.py -v