Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for getDeviceName (0.13 sec)

  1. tensorflow/c/eager/c_api_test.cc

      TF_Tensor* t = TFE_TensorHandleResolve(hcpu, status.get());
      ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    
      // Disable the test if no GPU is present.
      string gpu_device_name;
      if (GetDeviceName(ctx, &gpu_device_name, "GPU")) {
        TFE_TensorHandle* hgpu = TFE_TensorHandleCopyToDevice(
            hcpu, ctx, gpu_device_name.c_str(), status.get());
        ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
Back to top