Function CheckIndex

tensorflow/core/common_runtime/partitioning_utils_test.cc:170–175

}

void CheckIndex(const Node& node, int expected_index) {
  const AttrValue* attr_value;
  TF_ASSERT_OK(node.attrs().Find("index", &attr_value));
  int index = static_cast<int>(attr_value->i());
  ASSERT_EQ(expected_index, index);
}

TEST_F(PartitioningUtilsTest, UpdateArgsAndRets) {
  std::unique_ptr<Graph> graph = absl::make_unique<Graph>(OpRegistry::Global());