Searched refs:TFE_ContextListDevices (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/go/ |
| D | context.go | 103 devicesList := C.TFE_ContextListDevices(c.c, status.c)
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 102 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
|
| D | c_api_test_util.cc | 339 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetDeviceName()
|
| D | c_api_test.cc | 160 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
213 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenDevices()
330 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
|
| D | c_api_experimental_test.cc | 379 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
|
| D | c_api.cc | 167 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices() function
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 80 TFE_ContextListDevices(context.get(), status.get()), TF_DeleteDeviceList); in TEST()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | context.py | 481 device_list = pywrap_tfe.TFE_ContextListDevices(self._context_handle)
|
| /external/tensorflow/tensorflow/python/ |
| D | tfe_wrapper.cc | 636 auto output = TFE_ContextListDevices(tensorflow::InputTFE_Context(o), in PYBIND11_MODULE()
|