- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_DeviceListCount (0.08 sec)
-
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Aug 03 20:50:20 UTC 2023 - 94.6K bytes - Viewed (0) -
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Oct 26 21:08:15 UTC 2023 - 82.3K bytes - Viewed (0) -
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: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 16:27:48 UTC 2024 - 97K bytes - Viewed (0)