Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_DeviceListCount (0.41 sec)

  1. tensorflow/c/eager/c_api_test.cc

      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      TFE_DeleteContext(ctx);
      EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    
      const int num_devices = TF_DeviceListCount(devices);
      EXPECT_GE(num_devices, 1) << "At least one CPU device should exist";
      for (int i = 0; i < num_devices; ++i) {
        EXPECT_NE("", TF_DeviceListName(devices, i, status)) << i;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.h

    // Deallocates the device list.
    TF_CAPI_EXPORT extern void TF_DeleteDeviceList(TF_DeviceList* list);
    
    // Counts the number of elements in the device list.
    TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
    
    // Retrieves the full name of the device (e.g. /job:worker/replica:0/...)
    // The return value will be a pointer to a null terminated string. The caller
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

          TF_SessionListDevices(session, s), TF_DeleteDeviceList);
      TF_DeviceList* device_list = list.get();
    
      CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    
      const int num_devices = TF_DeviceListCount(device_list);
      LOG(INFO) << "There are " << num_devices << " devices.";
      for (int i = 0; i < num_devices; ++i) {
        const char* device_name = TF_DeviceListName(device_list, i, s);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
Back to top