- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TFE_DeleteContext (0.18 sec)
-
tensorflow/c/eager/c_api_test.cc
TFE_DeleteTensorHandle(handle_1); TF_DeleteStatus(status); TFE_DeleteTensorHandle(var_handle); } TFE_DeleteTensorHandle(handle_0); TFE_DeleteContext(ctx_0); TFE_DeleteContext(ctx_1); worker_server1.release(); worker_server2.release(); } TEST(CAPI, ShareVariableAcrossContextsAfterUpdateContextWorks) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
EXPECT_EQ(1.2f, value); TFE_DeleteTensorHandle(handle_1); TF_DeleteStatus(status); TFE_DeleteTensorHandle(var_handle); } TFE_DeleteTensorHandle(handle_0); TFE_DeleteContext(ctx_0); TFE_DeleteContext(ctx_1); worker_server1.release(); worker_server2.release(); } } // namespace
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/eager/c_api_distributed_test.cc
TFE_ExecutorWaitForAllPendingNodes(executor, status); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); TFE_DeleteExecutor(executor); TFE_ContextRemoveFunction(ctx, "AddVariablesFunction", status); TFE_DeleteContext(ctx); TF_DeleteStatus(status); // TODO(b/136478427): Figure out how to correctly shut the server down. worker_server1.release(); worker_server2.release(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/dlpack_test.cc
TestHandleFromDLPack(status, ctx, {4, 0, 2}, {0, 1, 1}); TestHandleFromDLPack(status, ctx, {4, 0, 2}, {0, 0, 1}); TestHandleFromDLPack(status, ctx, {4, 0, 2}, {0, 2, 0}); TFE_DeleteContext(ctx); TF_DeleteStatus(status); } } // namespace
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
std::make_unique<tensorflow::EagerContextDistributedManager>( eager_context)); #endif // !IS_MOBILE_PLATFORM return tensorflow::wrap(eager_context); } void TFE_DeleteContext(TFE_Context* ctx) { if (ctx == nullptr) { return; } // ctx->RefCountIsOne() should be true here. tensorflow::unwrap(ctx)->Release(); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2)