[BUG] Shape Func of Split Op Error

[wangxiang2713]

Function split_shape_func converts indices_or_sections to a list if indices_or_sections is actually indices (a int num):

if isinstance(indices_or_sections, int):
    indices_or_sections = [indices_or_sections]

and remains indices_or_sections a list if indices_or_sections is actually sections.

Then funtion _split_shape_func treats indices_or_sections as indices if len(indices_or_sections) == 1, treats indices_or_sections as sections otherwise:

if len(indices_or_sections) == 1:
    ...
else:
    ...

However, if indices_or_sections is actually sections but only contains one element, it will be treated as indices in error.

[vinx13]

Thanks for the PR. Could you add a test case?

[wangxiang2713]

Thanks for the PR. Could you add a test case?

Hi, a test case just added. This case may occur when convert a tensorflow model. As this code will be excute:

size_splits = _get_param(params, inputs[1])
section_beginnings = np.cumsum(size_splits)[:-1]
indices_or_sections = tuple(section_beginnings)

np.cumsum may convert a two elements list to one elements list.

[masahi]

I confirmed that the new tests work with your fix, thanks @wangxiang2713