Searched refs:TFE_ContextListDevices (Results 1 – 8 of 8) sorted by relevance
103 devicesList := C.TFE_ContextListDevices(c.c, status.c)
102 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
211 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetDeviceName()
136 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()183 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenDevices()300 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
513 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
746 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices() function
362 auto output = TFE_ContextListDevices(tensorflow::InputTFE_Context(o), in PYBIND11_MODULE()
468 device_list = pywrap_tfe.TFE_ContextListDevices(self._context_handle)