Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 2 of 2 for TFE_ContextSetThreadLocalDevicePlacementPolicy (0.58 seconds)

  1. 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)
  2. 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)
Back to Top