()
| 361 | |
| 362 | |
| 363 | def test_nms_wrong_valid_count_ndim(): |
| 364 | bb = relax.BlockBuilder() |
| 365 | data = relax.Var("data", R.Tensor((2, 10, 6), "float32")) |
| 366 | valid_count = relax.Var("valid_count", R.Tensor((2, 1), "int32")) |
| 367 | indices = relax.Var("indices", R.Tensor((2, 10), "int32")) |
| 368 | with pytest.raises(ValueError): |
| 369 | bb.normalize(relax.op.vision.non_max_suppression(data, valid_count, indices)) |
| 370 | |
| 371 | |
| 372 | def test_nms_wrong_indices_ndim(): |