From 5cf4311435e9087e0e9c7f4e1d4b415de6761530 Mon Sep 17 00:00:00 2001 From: Yujing Zhang Date: Thu, 14 May 2020 20:59:56 -0700 Subject: [PATCH] Fix a memory leak. PiperOrigin-RevId: 311662668 Change-Id: I59f9c9cdb8baed7a9828bb818ce1d293d185e6b6 --- tensorflow/c/eager/c_api_remote_test.cc | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/c/eager/c_api_remote_test.cc b/tensorflow/c/eager/c_api_remote_test.cc index 544dffb664c..d04e4ef4212 100644 --- a/tensorflow/c/eager/c_api_remote_test.cc +++ b/tensorflow/c/eager/c_api_remote_test.cc @@ -447,6 +447,9 @@ void VarIsInitialized(TFE_Context* ctx, TFE_TensorHandle* var_handle) { bool initialized = false; memcpy(&initialized, TF_TensorData(t), TF_TensorByteSize(t)); EXPECT_EQ(initialized, true); + TF_DeleteTensor(t); + TFE_DeleteTensorHandle(is_initialized[0]); + TFE_DeleteOp(op); delete status; }