Searched refs:TF_DeviceListCount (Results 1 – 12 of 12) sorted by relevance
47 size = tf_session.TF_DeviceListCount(raw_device_list)
735 size = tf_session.TF_DeviceListCount(raw_device_list)
1050 [](const TF_DeviceList* list) { return TF_DeviceListCount(list); }); in PYBIND11_MODULE()
214 const int num_devices = TF_DeviceListCount(devices); in GetDeviceName()
142 const int num_devices = TF_DeviceListCount(devices); in TEST()185 const int num_devices = TF_DeviceListCount(devices); in TensorHandleCopyBetweenDevices()302 const int num_devices = TF_DeviceListCount(devices); in TensorHandleCopyBetweenTwoGPUDevices()
515 int num_devices = TF_DeviceListCount(devices); in TEST()
87 for i := 0; i < int(C.TF_DeviceListCount(list)); i++ {
1386 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
603 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount() function
76 const int num_devices = TF_DeviceListCount(device_list); in GPUDeviceName()
774 m.def("TF_DeviceListCount", &TF_DeviceListCount); in PYBIND11_MODULE()
471 for i in range(pywrap_tfe.TF_DeviceListCount(device_list)):