- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TFE_ContextClearCaches (0.27 sec)
-
tensorflow/c/eager/c_api.h
TF_Status* status); // Clears the internal caches in the TFE context. Useful when reseeding random // ops. TF_CAPI_EXPORT extern void TFE_ContextClearCaches(TFE_Context* ctx); // Sets a thread-local device placement policy. After this call, other calls to // TFE_Execute in the same thread will use the device policy specified here
Registered: Tue Oct 29 12:39:09 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
} TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { TF_DeviceList* l = new TF_DeviceList; tensorflow::unwrap(ctx)->ListDevices(&l->response); return l; } void TFE_ContextClearCaches(TFE_Context* ctx) { tensorflow::unwrap(ctx)->ClearCachesAndThreadExecutors(); } // Set server_def on the context, possibly updating it. TF_CAPI_EXPORT extern void TFE_ContextSetServerDef(TFE_Context* ctx,
Registered: Tue Oct 29 12:39:09 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 43.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
status); CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); for (bool clear_cache : {true, false, true}) { if (clear_cache) { TFE_ContextClearCaches(ctx); } TFE_TensorHandle* m = TestMatrixTensorHandle(ctx); TFE_TensorHandle* retval[1] = {nullptr}; int num_retvals = 1; TFE_Op* op = TFE_NewOp(ctx, "MatMulFunction", status);
Registered: Tue Oct 29 12:39:09 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)