Fix a memory leak.
PiperOrigin-RevId: 311662668 Change-Id: I59f9c9cdb8baed7a9828bb818ce1d293d185e6b6
This commit is contained in:
parent
97f2fffe7e
commit
5cf4311435
@ -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;
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user