Searched refs:TF_DeviceList (Results 1 – 7 of 7) sorted by relevance
/external/tensorflow/tensorflow/c/ |
D | c_api.h | 1460 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*);
|
D | c_api_internal.h | 156 struct TF_DeviceList { struct
|
D | c_api.cc | 910 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, \
|
D | c_api_test.cc | 66 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/ |
D | c_api.h | 92 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
|
D | c_api_test.cc | 100 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()
|
D | c_api.cc | 118 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
|