Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for TF_DeviceListName (0.21 sec)

  1. tensorflow/c/eager/c_api_test_util.cc

      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());
        const string dev_name(TF_DeviceListName(devices, i, status.get()));
        CHECK_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get());
        if (dev_type == device_type) {
          *device_name = dev_name;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Wed Feb 21 22:37:46 UTC 2024
    - 23.5K bytes
    - Viewed (0)
  2. tensorflow/c/eager/c_api_test.cc

        TF_DeleteTensor(t);
        TFE_DeleteTensorHandle(hcpu);
        TFE_DeleteContext(ctx);
        return;
      }
      const string gpu_1_name(TF_DeviceListName(devices, 1, status.get()));
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK);
      const string gpu_2_name(TF_DeviceListName(devices, 2, status.get()));
      ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK);
      TFE_TensorHandle* hdevice =
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 20:50:20 UTC 2023
    - 94.6K bytes
    - Viewed (0)
  3. tensorflow/c/eager/c_api_experimental_test.cc

      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);
        TFE_TensorHandle* copy = TFE_TensorHandleCopyToDevice(m, ctx, name, status);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Aug 03 03:14:26 UTC 2023
    - 31.5K bytes
    - Viewed (0)
  4. tensorflow/c/c_api.h

    // TF_DeleteDeviceList.
    //
    // If index is out of bounds, an error code will be set in the status object,
    // and a null pointer will be returned.
    TF_CAPI_EXPORT extern const char* TF_DeviceListName(const TF_DeviceList* list,
                                                        int index,
                                                        TF_Status* status);
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  5. tensorflow/c/c_api.cc

        status->status = ::tensorflow::OkStatus();                            \
        return list->response[index].accessor;                                \
      }
    
    TF_DEVICELIST_METHOD(const char*, TF_DeviceListName, name().c_str(), nullptr);
    TF_DEVICELIST_METHOD(const char*, TF_DeviceListType, device_type().c_str(),
                         nullptr);
    TF_DEVICELIST_METHOD(int64_t, TF_DeviceListMemoryBytes, memory_limit(), -1);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      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);
        CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
        const char* device_type = TF_DeviceListType(device_list, i, s);
        CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s);
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top