Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/c/
Dc_api.h1375 typedef struct TF_DeviceList TF_DeviceList; typedef
1381 TF_CAPI_EXPORT extern TF_DeviceList* TF_SessionListDevices(TF_Session* session,
1388 TF_CAPI_EXPORT extern TF_DeviceList* TF_DeprecatedSessionListDevices(
1392 TF_CAPI_EXPORT extern void TF_DeleteDeviceList(TF_DeviceList* list);
1395 TF_CAPI_EXPORT extern int TF_DeviceListCount(const TF_DeviceList* list);
1404 TF_CAPI_EXPORT extern const char* TF_DeviceListName(const TF_DeviceList* list,
1415 TF_CAPI_EXPORT extern const char* TF_DeviceListType(const TF_DeviceList* list,
1424 const TF_DeviceList* list, int index, TF_Status* status);
1431 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()
597 TF_DeviceList* TF_DeprecatedSessionListDevices(TF_DeprecatedSession* session, in TF_DeprecatedSessionListDevices()
599 TF_DeviceList* response = new TF_DeviceList; in TF_DeprecatedSessionListDevices()
605 int TF_DeviceListCount(const TF_DeviceList* list) { in TF_DeviceListCount()
610 return_type method_name(const TF_DeviceList* list, const int index, \
Dc_api_test.cc72 std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> list( in GPUDeviceName()
74 TF_DeviceList* device_list = list.get(); in GPUDeviceName()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc95 PYBIND11_MAKE_OPAQUE(TF_DeviceList);
1060 TF_DeviceList* output = TF_SessionListDevices(session, status.get()); in PYBIND11_MODULE()
1067 [](const TF_DeviceList* list) { return TF_DeviceListCount(list); }); in PYBIND11_MODULE()
1069 m.def("TF_DeviceListName", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1077 m.def("TF_DeviceListType", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1085 m.def("TF_DeviceListMemoryBytes", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1093 m.def("TF_DeviceListIncarnation", [](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.cc167 TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { in TFE_ContextListDevices()
168 TF_DeviceList* l = new TF_DeviceList; in TFE_ContextListDevices()
Dc_api_test_util.cc339 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in GetDeviceName()
Dc_api_test.cc160 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
213 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenDevices()
330 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status.get()); in TensorHandleCopyBetweenTwoGPUDevices()
Dc_api_experimental_test.cc379 TF_DeviceList* devices = TFE_ContextListDevices(ctx, status); in TEST()
/external/tensorflow/tensorflow/python/
Dtfe_wrapper.cc75 PYBIND11_MAKE_OPAQUE(TF_DeviceList);
516 py::class_<TF_DeviceList> TF_DeviceList_class(m, "TF_DeviceList"); in PYBIND11_MODULE()
1083 m.def("TF_DeviceListName", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
1090 m.def("TF_DeviceListType", [](const TF_DeviceList* list, int index) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/go/
Dsession.go84 func deviceSliceFromDeviceList(list *C.TF_DeviceList) ([]Device, error) { argument
/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc79 std::unique_ptr<TF_DeviceList, decltype(&TF_DeleteDeviceList)> devices( in TEST()