0a2e8beac36bf6c797bf45cdfa27cdd6d2924796,onnx_tf/backend.py,TensorflowBackend,handle_add,#Any#Any#Any#,390

Before Change

  @classmethod
  def handle_add(cls, node, input_dict):
    x = input_dict[node.inputs[0]]
    y = input_dict[node.inputs[1]]
    if "broadcast" in node.attrs.keys():
      broadcast = node.attrs["broadcast"]
    else:
      broadcast = 0
    if broadcast == 0:
      warnings.warn("Definition of Add with broadcast disabled is incompatible"
                    "between onnx and tensorflow.", UserWarning)
    if "axis" in node.attrs.keys():
      warnings.warn("Unsupported axis attribute by Tensorflow in Add."
                    "This attribute will be ignored.", UserWarning)
    return [tf.add(x, cls._explicit_broadcast(y, broadcast))]

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

After Change

  @classmethod
  def handle_add(cls, node, input_dict):
    return [cls._bin_op(node, input_dict, tf.add)]

  @classmethod
  def handle_arg_max(cls, node, input_dict):
    data = input_dict[node.inputs[0]]

Instances