Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api.h1460 typedef struct TF_DeviceList TF_DeviceList; typedef
1466 TF_CAPI_EXPORT extern TF_DeviceList* TF_SessionListDevices(TF_Session* session,
1473 TF_CAPI_EXPORT extern TF_DeviceList* TF_DeprecatedSessionListDevices(
1477 TF_CAPI_EXPORT extern void TF_DeleteDeviceList(TF_DeviceList* list);
1480 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
1489 TF_CAPI_EXPORT extern const char* TF_DeviceListName(const TF_DeviceList* list,
1499 TF_CAPI_EXPORT extern const char* TF_DeviceListType(const TF_DeviceList* list,
1507 const TF_DeviceList* list, int index, TF_Status*);
Dc_api_internal.h156 struct TF_DeviceList { struct
Dc_api.cc910 void TF_DeleteDeviceList(TF_DeviceList* s) { delete s; } in TF_DeleteDeviceList()
912 TF_DeviceList* TF_SessionListDevices(TF_Session* session, TF_Status* status) { in TF_SessionListDevices()
913 TF_DeviceList* response = new TF_DeviceList; in TF_SessionListDevices()
918 TF_DeviceList* TF_DeprecatedSessionListDevices(TF_DeprecatedSession* session, in TF_DeprecatedSessionListDevices()
920 TF_DeviceList* response = new TF_DeviceList; in TF_DeprecatedSessionListDevices()
925 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount()
930 return_type method_name(const TF_DeviceList* list, const int index, \
Dc_api_test.cc66 std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> list( in GPUDeviceName()
68 TF_DeviceList* device_list = list.get(); in GPUDeviceName()
/external/tensorflow/tensorflow/c/eager/
Dc_api.h92 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
Dc_api_test.cc100 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetGPUDeviceName()
174 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
220 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TEST()
289 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TEST()
Dc_api.cc118 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()