Search Options

Display Count
Sort
Preferred Language
Advanced Search

Results 1 - 6 of 6 for TF_DeviceListCount (0.19 seconds)

  1. tensorflow/c/eager/c_api_test_util.cc

          TF_NewStatus(), TF_DeleteStatus);
      TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get());
      CHECK_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get());
    
      const int num_devices = TF_DeviceListCount(devices);
      for (int i = 0; i < num_devices; ++i) {
        const string dev_type(TF_DeviceListType(devices, i, status.get()));
        CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 23.4K bytes
    - Click Count (0)
  2. tensorflow/c/eager/c_api_experimental_test.cc

      float* m_float = static_cast<float*>(TF_TensorData(m_data));
      TF_DeviceList* devices = TFE_ContextListDevices(ctx, status);
      CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
      int num_devices = TF_DeviceListCount(devices);
      for (int d = 0; d < num_devices; ++d) {
        const char* name = TF_DeviceListName(devices, d, status);
        CHECK_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 09 05:56:18 GMT 2025
    - 31.5K bytes
    - Click Count (0)
  3. 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;
    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)
  4. 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
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Click Count (0)
  5. tensorflow/c/c_api.cc

      TF_DeviceList* response = new TF_DeviceList;
      if (session && session->session)
        status->status = session->session->ListDevices(&response->response);
      return response;
    }
    
    int TF_DeviceListCount(const TF_DeviceList* list) {
      return list->response.size();
    }
    
    #define TF_DEVICELIST_METHOD(return_type, method_name, accessor, err_val) \
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Sat Oct 04 05:55:32 GMT 2025
    - 102.4K bytes
    - Click Count (0)
  6. 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);
    Created: Tue Apr 07 12:39:13 GMT 2026
    - Last Modified: Wed Jan 07 04:56:09 GMT 2026
    - 97.3K bytes
    - Click Count (0)
Back to Top