- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for device_type (0.05 sec)
-
tensorflow/c/c_api_experimental.cc
} void TF_AttrBuilderCheckCanRunOnDevice(TF_AttrBuilder* builder, const char* device_type, TF_Status* status) { status->status = tensorflow::FindKernelDef( tensorflow::DeviceType(device_type), builder->BuildNodeDef(), /* def = */ nullptr, /* kernel_class_name = */ nullptr); }
Registered: Tue Sep 09 12:39:10 UTC 2025 - Last Modified: Mon Aug 18 03:53:25 UTC 2025 - 29.5K bytes - Viewed (0) -
tensorflow/c/c_api_test.cc
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); LOG(INFO) << "Device " << i << " has name " << device_name << ", type " << device_type; if (string(device_type) == DEVICE_GPU) { return device_name; } } // No GPU device found. return ""; }
Registered: Tue Sep 09 12:39:10 UTC 2025 - Last Modified: Fri Dec 27 12:18:10 UTC 2024 - 97K bytes - Viewed (0) -
tensorflow/c/c_api.cc
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); TF_DEVICELIST_METHOD(uint64_t, TF_DeviceListIncarnation, incarnation(), 0);
Registered: Tue Sep 09 12:39:10 UTC 2025 - Last Modified: Tue May 13 06:30:43 UTC 2025 - 102.3K bytes - Viewed (0)