c4fc047e7ec71daa6aa8f71e9cc2ee9e5a3768b6,onnx_tf/handlers/backend/split.py,Split,_common,#Any#Any#,26

Before Change

  @classmethod
  def _common(cls, node, **kwargs):
    tensor_dict = kwargs["tensor_dict"]
    x_shape = tensor_dict[node.inputs[0]].get_shape().as_list()
    attrs = copy.deepcopy(node.attrs)
    axis = attrs.get("axis", 0)
    axis = axis if axis >= 0 else len(x_shape) + axis
    if "split" in node.attrs:

After Change

      split = tensor_dict[node.inputs[1]]
    else:
      per_part = x_shape[axis] / len(node.outputs)
      if input.shape.is_fully_defined():
        if int(per_part) != per_part:
          raise ValueError("Split can not be evenly divided.")
        split = [int(per_part)] * len(node.outputs)
      else:
        split = [tf.cast(per_part, tf.int32)] * len(node.outputs)
    attrs["num_or_size_splits"] = split
    return list(
        cls.make_tensor_from_onnx_node(
            node, inputs=[input], attrs=attrs, **kwargs))

Instances