From d371ce10cfd965be7166affc9f56e0e1a4bffa70 Mon Sep 17 00:00:00 2001
From: Fei Hu <hufei68@gmail.com>
Date: Tue, 19 Mar 2019 16:22:48 -0700
Subject: [PATCH] Add a missing release of concatenate_dataset

---
 tensorflow/core/kernels/data/concatenate_dataset_op_test.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tensorflow/core/kernels/data/concatenate_dataset_op_test.cc b/tensorflow/core/kernels/data/concatenate_dataset_op_test.cc
index 6517719e9bb..554b31c9103 100644
--- a/tensorflow/core/kernels/data/concatenate_dataset_op_test.cc
+++ b/tensorflow/core/kernels/data/concatenate_dataset_op_test.cc
@@ -384,6 +384,7 @@ TEST_F(ConcatenateDatasetOpTest, DatasetSave) {
   DatasetBase *concatenate_dataset;
   TF_ASSERT_OK(CreateDataset(dataset_kernel.get(), dataset_kernel_ctx.get(),
                              &concatenate_dataset));
+  core::ScopedUnref scored_unref(concatenate_dataset);
 
   std::unique_ptr<SerializationContext> serialization_ctx;
   TF_ASSERT_OK(CreateSerializationContext(&serialization_ctx));