Home
last modified time | relevance | path

Searched refs:TF_Library (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Dtensorflow_jni.cc41 TF_Library* h = TF_LoadLibrary(cname, status); in Java_org_tensorflow_TensorFlow_libraryLoad()
51 TF_DeleteLibraryHandle(reinterpret_cast<TF_Library*>(handle)); in Java_org_tensorflow_TensorFlow_libraryDelete()
57 TF_Buffer buf = TF_GetOpList(reinterpret_cast<TF_Library*>(handle)); in Java_org_tensorflow_TensorFlow_libraryOpList()
/external/tensorflow/tensorflow/c/
Dc_api_experimental.h324 TF_CAPI_EXPORT extern TF_Library* TF_LoadPluggableDeviceLibrary(
330 TF_Library* lib_handle);
Dc_api.h1452 typedef struct TF_Library TF_Library; typedef
1465 TF_CAPI_EXPORT extern TF_Library* TF_LoadLibrary(const char* library_filename,
1473 TF_CAPI_EXPORT extern TF_Buffer TF_GetOpList(TF_Library* lib_handle);
1477 TF_CAPI_EXPORT extern void TF_DeleteLibraryHandle(TF_Library* lib_handle);
Dc_api_experimental.cc725 TF_Library* TF_LoadPluggableDeviceLibrary(const char* library_filename, in TF_LoadPluggableDeviceLibrary()
732 TF_Library* lib_handle = new TF_Library; in TF_LoadPluggableDeviceLibrary()
758 void TF_DeletePluggableDeviceLibraryHandle(TF_Library* lib_handle) { in TF_DeletePluggableDeviceLibraryHandle()
Dc_api_internal.h65 struct TF_Library { struct
Dc_api_experimental_test.cc248 TF_Library* lib = TF_LoadPluggableDeviceLibrary(lib_path.c_str(), status); in TEST()
Dc_api.cc553 TF_Library* TF_LoadLibrary(const char* library_filename, TF_Status* status) { in TF_LoadLibrary()
554 TF_Library* lib_handle = new TF_Library; in TF_LoadLibrary()
565 TF_Buffer TF_GetOpList(TF_Library* lib_handle) { return lib_handle->op_list; } in TF_GetOpList()
567 void TF_DeleteLibraryHandle(TF_Library* lib_handle) { in TF_DeleteLibraryHandle()
Dc_api_test.cc204 TF_Library* lib = TF_LoadLibrary(lib_path.c_str(), status); in TEST()
/external/tensorflow/tensorflow/python/client/
Dtf_session_wrapper.cc98 PYBIND11_MAKE_OPAQUE(TF_Library);
131 py::class_<TF_Library> TF_Library_class(m, "TF_Library"); in PYBIND11_MODULE()
752 m.def("TF_GetOpList", [](TF_Library* lib_handle) { in PYBIND11_MODULE()
/external/tensorflow/tensorflow/go/
Dgraph.go500 cptr *C.TF_Library