()
| 130 | |
| 131 | |
| 132 | def test_roi_align_wrong_layout(): |
| 133 | bb = relax.BlockBuilder() |
| 134 | x = relax.Var("x", R.Tensor((2, 3, 32, 32), "float32")) |
| 135 | rois = relax.Var("rois", R.Tensor((4, 5), "float32")) |
| 136 | |
| 137 | with pytest.raises(ValueError): |
| 138 | bb.normalize(relax.op.vision.roi_align(x, rois, (7, 7), 1.0, layout="HWCN")) |
| 139 | |
| 140 | |
| 141 | def test_roi_align_legalize(): |