- Sort Score
- Num 10 results
- Language All
Results 1 - 2 of 2 for TFE_ContextSetThreadLocalDevicePlacementPolicy (0.63 seconds)
-
tensorflow/c/eager/c_api.cc
status->status = tensorflow::OkStatus(); #else // !defined(IS_MOBILE_PLATFORM) status->status = tensorflow::unwrap(ctx)->AsyncWait(); #endif // !IS_MOBILE_PLATFORM } void TFE_ContextSetThreadLocalDevicePlacementPolicy( TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy) { tensorflow::unwrap(ctx)->SetThreadLocalDevicePlacementPolicy( static_cast<tensorflow::ContextDevicePlacementPolicy>(policy)); }
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Fri Nov 07 05:55:21 GMT 2025 - 43.9K bytes - Click Count (0) -
tensorflow/c/eager/c_api_test.cc
TFE_ContextOptionsSetDevicePlacementPolicy(opts, global_policy); TFE_Context* ctx = TFE_NewContext(opts, status.get()); if (thread_policy != global_policy) { TFE_ContextSetThreadLocalDevicePlacementPolicy(ctx, thread_policy); } TFE_DeleteContextOptions(opts); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); TFE_TensorHandle* hcpu = TestMatrixTensorHandle(ctx);
Created: Tue Apr 07 12:39:13 GMT 2026 - Last Modified: Thu Oct 09 05:56:18 GMT 2025 - 94.6K bytes - Click Count (0)