Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api.h1366 typedef struct TF_DeviceList TF_DeviceList; typedef
1372 TF_CAPI_EXPORT extern TF_DeviceList* TF_SessionListDevices(TF_Session* session,
1379 TF_CAPI_EXPORT extern TF_DeviceList* TF_DeprecatedSessionListDevices(
1383 TF_CAPI_EXPORT extern void TF_DeleteDeviceList(TF_DeviceList* list);
1386 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
1395 TF_CAPI_EXPORT extern const char* TF_DeviceListName(const TF_DeviceList* list,
1406 TF_CAPI_EXPORT extern const char* TF_DeviceListType(const TF_DeviceList* list,
1415 const TF_DeviceList* list, int index, TF_Status* status);
1422 const TF_DeviceList* list, int index, TF_Status* status);
Dc_api_internal.h154 struct TF_DeviceList { struct
Dc_api.cc588 void TF_DeleteDeviceList(TF_DeviceList* list) { delete list; } in TF_DeleteDeviceList()
590 TF_DeviceList* TF_SessionListDevices(TF_Session* session, TF_Status* status) { in TF_SessionListDevices()
591 TF_DeviceList* response = new TF_DeviceList; in TF_SessionListDevices()
596 TF_DeviceList* TF_DeprecatedSessionListDevices(TF_DeprecatedSession* session, in TF_DeprecatedSessionListDevices()
598 TF_DeviceList* response = new TF_DeviceList; in TF_DeprecatedSessionListDevices()
603 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount()
608 return_type method_name(const TF_DeviceList* list, const int index, \
Dc_api_test.cc70 std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> list( in GPUDeviceName()
72 TF_DeviceList* device_list = list.get(); in GPUDeviceName()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc95 PYBIND11_MAKE_OPAQUE(TF_DeviceList);
1043 TF_DeviceList* output = TF_SessionListDevices(session, status.get()); in PYBIND11_MODULE()
1050 [](const TF_DeviceList* list) { return TF_DeviceListCount(list); }); in PYBIND11_MODULE()
1052 m.def("TF_DeviceListName", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1060 m.def("TF_DeviceListType", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1068 m.def("TF_DeviceListMemoryBytes", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1076 m.def("TF_DeviceListIncarnation", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc68 PYBIND11_MAKE_OPAQUE(TF_DeviceList);
323 py::class_<TF_DeviceList> TF_DeviceList_class(m, "TF_DeviceList"); in PYBIND11_MODULE()
775 m.def("TF_DeviceListName", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
782 m.def("TF_DeviceListType", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/c/eager/
Dc_api.h102 TF_CAPI_EXPORT extern TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx,
Dc_api_test_util.cc211 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetDeviceName()
Dc_api.cc746 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
747 TF_DeviceList* l = new TF_DeviceList; in TFE_ContextListDevices()
Dc_api_test.cc136 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()
Dc_api_experimental_test.cc513 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
/external/tensorflow/tensorflow/go/
Dsession.go83 func deviceSliceFromDeviceList(list *C.TF_DeviceList) ([]Device, error) { argument