()
| 352 | |
| 353 | |
| 354 | def test_nms_wrong_ndim(): |
| 355 | bb = relax.BlockBuilder() |
| 356 | data = relax.Var("data", R.Tensor((10, 6), "float32")) |
| 357 | valid_count = relax.Var("valid_count", R.Tensor((2,), "int32")) |
| 358 | indices = relax.Var("indices", R.Tensor((2, 10), "int32")) |
| 359 | with pytest.raises(ValueError): |
| 360 | bb.normalize(relax.op.vision.non_max_suppression(data, valid_count, indices)) |
| 361 | |
| 362 | |
| 363 | def test_nms_wrong_valid_count_ndim(): |