()
| 261 | |
| 262 | |
| 263 | def test_get_valid_counts_wrong_ndim(): |
| 264 | bb = relax.BlockBuilder() |
| 265 | data = relax.Var("data", R.Tensor((10, 6), "float32")) |
| 266 | with pytest.raises(ValueError): |
| 267 | bb.normalize(relax.op.vision.get_valid_counts(data)) |
| 268 | |
| 269 | |
| 270 | def test_get_valid_counts_invalid_indices(): |