Home
last modified time | relevance | path

Searched refs:TF_DeviceListCount (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/client/
Dsession_list_devices_test.py47 size = tf_session.TF_DeviceListCount(raw_device_list)
Dsession.py706 size = tf_session.TF_DeviceListCount(raw_device_list)
/external/tensorflow/tensorflow/c/eager/
Dc_api_test_util.cc183 const int num_devices = TF_DeviceListCount(devices); in GetDeviceName()
Dc_api_test.cc103 const int num_devices = TF_DeviceListCount(devices); in TEST()
474 const int num_devices = TF_DeviceListCount(devices); in TensorHandleCopyBetweenDevices()
588 const int num_devices = TF_DeviceListCount(devices); in TensorHandleCopyBetweenTwoGPUDevices()
/external/tensorflow/tensorflow/go/
Dsession.go87 for i := 0; i < int(C.TF_DeviceListCount(list)); i++ {
/external/tensorflow/tensorflow/python/eager/
Dcontext.py320 for i in range(pywrap_tensorflow.TF_DeviceListCount(device_list)):
/external/tensorflow/tensorflow/c/
Dc_api.h1599 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
Dc_api_test.cc75 const int num_devices = TF_DeviceListCount(device_list); in GPUDeviceName()
Dc_api.cc1027 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount() function