x = constant_op.constant([1], name='x_const')
  y = constant_op.constant([2], name='y_const')
  math_ops.add(x, y, name='x_y_sum')


def tfadd_with_ckpt(out_dir):
  x = array_ops.placeholder(dtypes.int32, name='x_hold')
  y = variable_v1.VariableV1(constant_op.constant([0]), name='y_saved')
  math_ops.add(x, y, name='x_y_sum')

  init_op = variables.global_variables_initializer()
  saver = saver_lib.Saver(write_version=saver_pb2.SaverDef.V1)

  x = value
  y = value
  math_ops.add(x, y, name='x_y_sum')

traces: {
    key  : "x@"
    value: {
      file_line_cols: {
        line      : 1
        col       : 1
      }
    }
}
traces: {
    key  : "x_y_sum@"
    value: {
      file_line_cols: {
        line      : 3
        col       : 1
      }
      file_line_cols: {
        line      : 4
        col       : 1
      }
    }
}
traces: {

# Text form of tensorflow.tf2xla.Config proto.
feed {
  id { node_name: "x_hold" }
  shape {
    dim { size: 1 }
  }
}
fetch {
  id { node_name: "x_y_sum" }

feed {
  id { node_name: "x" }
  shape {
    dim { size: 2 }
  }
}
feed {
  id { node_name: "y" }
  shape {
    dim { size: 3 }
  }
}
fetch {
  id { node_name: "x_y_sum" }

feed {
  id { node_name: "x_const" }
  shape {
    dim { size: 1 }
  }
}
feed {
  id { node_name: "y_const" }
  shape {
    dim { size: 1 }
  }
}
fetch {
  id { node_name: "x_y_sum" }

  op    : "Reshape"
  input : "y_const"
  input : "y_shape"
  attr { key: "T" value { type: DT_INT32 } }
  # Attribute TShape not specified; needs to be set to its default
  # by tfcompile.
}
node {
  name  : "x_y_sum"
  op    : "Add"
  input : "x_const"
  input : "y_reshape"
  attr {
    key  : "T"
    value {
      type: DT_INT32
    }
  }
}
versions {
  producer: 15

Composite = composite.Composite

#--- test fn for mlir location ---


@Composite('TestInputNOp')
def _tfr_loc_test(x):
  n = 10
  x_sum = x[0]
  for i in range(1, n):
    x_sum = math_ops.Add(x_sum, x[i])
  return x_sum


#--- test fn for tfr tensors ---


@composite.Composite('TestNoOp')
def _tfr_tensor_empty_arg():
  pass


@composite.Composite('TestIdentityOp')

// to allow a Python debugger to insert source information about the graph node.
// For example, a Python add expression may be represented as
// {{node, x_y_sum}} = Add(x, y) in the error message. See routine interpolate
// in tensorflow/python/framework/error_interpolation.py for more detail.
static std::string InterpolateErrorMessage(std::string message) {

  EXPECT_EQ(muladd.LookupArgIndex("y_hold"), -1);
  EXPECT_EQ(muladd.LookupArgIndex("x_y_prod"), -1);
  EXPECT_EQ(muladd.LookupArgIndex("x_y_sum"), -1);

  EXPECT_EQ(muladd.LookupResultIndex("x_y_prod"), 0);
  EXPECT_EQ(muladd.LookupResultIndex("x_y_sum"), 1);
  EXPECT_EQ(muladd.LookupResultIndex(""), -1);
  EXPECT_EQ(muladd.LookupResultIndex("x"), -1);
  EXPECT_EQ(muladd.LookupResultIndex("y"), -1);