eddede8a34d5a25c5aadddd8a285df808cca1d91,onnx_tf/backends/backend_v2.py,TensorflowBackend,handle_split,#Any#Any#Any#,53

Before Change

  @classmethod
  def handle_split(cls, node, input_dict):
    split = tf.constant(node.attrs["split"])
    axis = node.attrs["axis"]
    return list(tf.split(input_dict[node.inputs[0]], split, axis))

After Change

    if "split" in node.attrs:
      split = node.attrs["split"]
    else:
      per_part = x_shape[axis] / len(node.outputs)
      assert int(per_part) == per_part
      split = [int(per_part)] * len(node.outputs)
    return list(tf.split(input_dict[node.inputs[0]], split, axis))

Instances