Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TFE_ContextDevicePlacementPolicy (0.41 sec)

  1. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Apr 27 21:07:00 UTC 2023
    - 22.8K bytes
    - Viewed (0)
  2. 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: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top