2a4e2b1b131d9a1b48d8eba5479b585bf328d3ba,onnx_tf/backend.py,TensorflowBackend,handle_elu,#Any#Any#Any#,745

Before Change

  @classmethod
  def handle_elu(cls, node, input_dict):
    x = input_dict[node.inputs[0]]
    if "alpha" in node.attrs.keys():
      warnings.warn("Unsupported alpha attribute by Tensorflow in Elu."
                    "This attribute will be ignored.", UserWarning)
    return [tf.nn.elu(x)]

  @classmethod
  def handle_flatten(cls, node, input_dict):

After Change

    x = input_dict[node.inputs[0]]

    alpha = node.attrs.get("alpha", 1.0)
    if "alpha" in node.attrs.keys():
      return [tf.cast(x < 0.0, tf.float32) * alpha * (tf.exp(x) - 1.0) + tf.cast(x >= 0.0, tf.float32) * x]
    else:
      return [tf.nn.elu(x)]

  @classmethod
  def handle_flatten(cls, node, input_dict):
    tensor = input_dict[node.inputs[0]]
    axis = node.attrs["axis"] if "axis" in node.attrs.keys() else 1

Instances