Fix a memory leak.

PiperOrigin-RevId: 311662668
Change-Id: I59f9c9cdb8baed7a9828bb818ce1d293d185e6b6
This commit is contained in:
Yujing Zhang 2020-05-14 20:59:56 -07:00 committed by TensorFlower Gardener
parent 97f2fffe7e
commit 5cf4311435

View File

@ -447,6 +447,9 @@ void VarIsInitialized(TFE_Context* ctx, TFE_TensorHandle* var_handle) {
bool initialized = false; bool initialized = false;
memcpy(&initialized, TF_TensorData(t), TF_TensorByteSize(t)); memcpy(&initialized, TF_TensorData(t), TF_TensorByteSize(t));
EXPECT_EQ(initialized, true); EXPECT_EQ(initialized, true);
TF_DeleteTensor(t);
TFE_DeleteTensorHandle(is_initialized[0]);
TFE_DeleteOp(op);
delete status; delete status;
} }