- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for TF_DeviceListName (0.08 sec)
-
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) -
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) -
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) -
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) -
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) -
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)