def METHOD_NAME(self):
    # Test that conv2d fails for dimensions other than 2 or 3.
    with pytest.raises(Exception):
        conv.conv2d(dtensor4(), dtensor3())
    with pytest.raises(Exception):
        conv.conv2d(dtensor3(), dvector())