Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api.h1579 typedef struct TF_DeviceList TF_DeviceList; typedef
1585 TF_CAPI_EXPORT extern TF_DeviceList* TF_SessionListDevices(TF_Session* session,
1592 TF_CAPI_EXPORT extern TF_DeviceList* TF_DeprecatedSessionListDevices(
1596 TF_CAPI_EXPORT extern void TF_DeleteDeviceList(TF_DeviceList* list);
1599 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
1608 TF_CAPI_EXPORT extern const char* TF_DeviceListName(const TF_DeviceList* list,
1619 TF_CAPI_EXPORT extern const char* TF_DeviceListType(const TF_DeviceList* list,
1628 const TF_DeviceList* list, int index, TF_Status* status);
1635 const TF_DeviceList* list, int index, TF_Status* status);
Dc_api_internal.h162 struct TF_DeviceList { struct
Dc_api.cc1012 void TF_DeleteDeviceList(TF_DeviceList* s) { delete s; } in TF_DeleteDeviceList()
1014 TF_DeviceList* TF_SessionListDevices(TF_Session* session, TF_Status* status) { in TF_SessionListDevices()
1015 TF_DeviceList* response = new TF_DeviceList; in TF_SessionListDevices()
1020 TF_DeviceList* TF_DeprecatedSessionListDevices(TF_DeprecatedSession* session, in TF_DeprecatedSessionListDevices()
1022 TF_DeviceList* response = new TF_DeviceList; in TF_DeprecatedSessionListDevices()
1027 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount()
1032 return_type method_name(const TF_DeviceList* list, const int index, \
Dc_api_test.cc69 std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> list( in GPUDeviceName()
71 TF_DeviceList* device_list = list.get(); in GPUDeviceName()
/external/tensorflow/tensorflow/c/eager/
Dc_api.h96 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
Dc_api.cc367 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
368 TF_DeviceList* list = new TF_DeviceList; in TFE_ContextListDevices()
Dc_api_test_util.cc180 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetDeviceName()
Dc_api_test.cc97 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
472 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenDevices()
586 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
/external/tensorflow/tensorflow/go/
Dsession.go83 func deviceSliceFromDeviceList(list *C.TF_DeviceList) ([]Device, error) { argument