aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/c/c_api_test.cc
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/c/c_api_test.cc')
-rw-r--r--tensorflow/c/c_api_test.cc23
1 files changed, 23 insertions, 0 deletions
diff --git a/tensorflow/c/c_api_test.cc b/tensorflow/c/c_api_test.cc
index c470ab5649..e674b1623c 100644
--- a/tensorflow/c/c_api_test.cc
+++ b/tensorflow/c/c_api_test.cc
@@ -1426,6 +1426,29 @@ TEST(CAPI, SavedModelNullArgsAreValid) {
TF_DeleteStatus(s);
}
+TEST(CAPI, DeletingNullPointerIsSafe) {
+ TF_Status* status = TF_NewStatus();
+
+ TF_DeleteStatus(nullptr);
+ TF_DeleteBuffer(nullptr);
+ TF_DeleteTensor(nullptr);
+ TF_DeleteSessionOptions(nullptr);
+ TF_DeleteGraph(nullptr);
+ TF_DeleteImportGraphDefOptions(nullptr);
+ TF_DeleteImportGraphDefResults(nullptr);
+ TF_DeleteFunction(nullptr);
+ TF_DeleteSession(nullptr, status);
+ EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
+ TF_DeletePRunHandle(nullptr);
+ TF_DeleteDeprecatedSession(nullptr, status);
+ EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
+ TF_DeleteDeviceList(nullptr);
+ TF_DeleteLibraryHandle(nullptr);
+ TF_DeleteApiDefMap(nullptr);
+
+ TF_DeleteStatus(status);
+}
+
REGISTER_OP("TestOpWithNoGradient")
.Input("x: T")
.Output("y: T")