- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for TFE_ContextDevicePlacementPolicy (0.37 sec)
-
tensorflow/c/eager/c_api.h
TF_CAPI_EXPORT extern void TFE_ContextSetThreadLocalDevicePlacementPolicy( TFE_Context* ctx, TFE_ContextDevicePlacementPolicy policy); // Returns the device placement policy to be used by this context in the current // thread. TF_CAPI_EXPORT extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy(TFE_Context* ctx);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_internal.h
// c_api_experimental.h struct TFE_ContextOptions { TF_SessionOptions session_options; // true if async execution is enabled. bool async = false; TFE_ContextDevicePlacementPolicy device_placement_policy{ TFE_DEVICE_PLACEMENT_SILENT}; // If true, use TFRT backend bool use_tfrt = false; // Whether to run elementary eager ops wrapped in a call op.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 18 19:26:34 UTC 2023 - 1.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
// the appropriate client thread. In particular, in async mode, it may not be // safe to call this function from the async EagerExecutor threads. extern TFE_ContextDevicePlacementPolicy TFE_ContextGetDevicePlacementPolicy( TFE_Context* ctx) { return static_cast<TFE_ContextDevicePlacementPolicy>( tensorflow::unwrap(ctx)->GetDevicePlacementPolicy()); } TFE_TensorHandle* TFE_NewTensorHandle(const TF_Tensor* t, TF_Status* status) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 08:11:23 UTC 2024 - 44K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test.cc
TEST(CAPI, TensorHandleCopyBetweenTwoGPUDevicesAsync) { TensorHandleCopyBetweenTwoGPUDevices(true); } void TensorHandleSilentCopy(bool async, TFE_ContextDevicePlacementPolicy global_policy, TFE_ContextDevicePlacementPolicy thread_policy, bool cpu_op) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0)