- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TFE_DeleteContext (0.16 sec)
-
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/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)