def METHOD_NAME(node, graph):
    data = node.inputs[0].get_tensor_value(as_list=False)
    axis = node.get_attr_value('axis', 0)
    if len(node.output) == 1:
        return [data]
    split = node.get_attr_value('split')
    if len(node.input) > 1:
        split = node.inputs[1].get_tensor_value(as_list=False)
    if split is not None:
        indices_or_sections = np.cumsum(split[:-1])
    else:
        indices_or_sections = len(node.output)
    return np.split(data, indices_or_sections, axis)