Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TFE_ContextSetThreadLocalDevicePlacementPolicy (0.18 sec)

  1. 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);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top