dim { size: 2 }
  }
  name: "x"
}
feed {
  id { node_name: "y_hold" }
  shape {
    dim { size: 2 }
    dim { size: 2 }
  }
  name: "y"
}
fetch {
  id { node_name: "x_y_prod" }
  name: "x_y_prod"
}
fetch {
  id { node_name: "x_y_sum" }
  name: "x_y_sum"

  shape {
    dim { size: 2 }
    dim { size: 3 }
  }
}
feed {
  id { node_name: "y_hold" }
  shape {
    dim { size: 3 }
    dim { size: 2 }
  }
}
fetch {
  id { node_name: "x_y_prod" }

  shape {
    dim { size: <M> }
    dim { size: <K> }
  }
}
feed {
  id { node_name: "y_hold" }
  shape {
    dim { size: <K> }
    dim { size: <N> }
  }
}
fetch {
  id { node_name: "x_y_prod" }

  y = array_ops.placeholder(dtypes.float32, name='y_hold')
  math_ops.matmul(x, y, name='x_y_prod')


def tfmatmulandadd(_):
  # This tests multiple outputs.
  x = array_ops.placeholder(dtypes.float32, name='x_hold')
  y = array_ops.placeholder(dtypes.float32, name='y_hold')
  math_ops.matmul(x, y, name='x_y_prod')
  math_ops.add(x, y, name='x_y_sum')


def tffunction(_):

  EXPECT_EQ(muladd.LookupArgIndex("x_hold"), -1);
  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);