Searched refs:lib_handle (Results 1 – 8 of 8) sorted by relevance
54 lib_handle = py_tf.TF_LoadLibrary(library_filename)57 py_tf.TF_GetOpList(lib_handle))61 py_tf.TF_DeleteLibraryHandle(lib_handle)
732 TF_Library* lib_handle = new TF_Library; in TF_LoadPluggableDeviceLibrary()741 lib_handle->lib_handle = it->second; in TF_LoadPluggableDeviceLibrary()744 env->LoadDynamicLibrary(library_filename, &lib_handle->lib_handle); in TF_LoadPluggableDeviceLibrary()747 tensorflow::RegisterPluggableDevicePlugin(lib_handle->lib_handle)); in TF_LoadPluggableDeviceLibrary()749 delete lib_handle; in TF_LoadPluggableDeviceLibrary()753 return lib_handle; in TF_LoadPluggableDeviceLibrary()758 void TF_DeletePluggableDeviceLibraryHandle(TF_Library* lib_handle) { in TF_DeletePluggableDeviceLibraryHandle() argument759 delete lib_handle; in TF_DeletePluggableDeviceLibraryHandle()
554 TF_Library* lib_handle = new TF_Library; in TF_LoadLibrary() local556 library_filename, &lib_handle->lib_handle, &lib_handle->op_list.data, in TF_LoadLibrary()557 &lib_handle->op_list.length); in TF_LoadLibrary()559 delete lib_handle; in TF_LoadLibrary()562 return lib_handle; in TF_LoadLibrary()565 TF_Buffer TF_GetOpList(TF_Library* lib_handle) { return lib_handle->op_list; } in TF_GetOpList() argument567 void TF_DeleteLibraryHandle(TF_Library* lib_handle) { in TF_DeleteLibraryHandle() argument568 if (lib_handle == nullptr) return; in TF_DeleteLibraryHandle()569 tensorflow::port::Free(const_cast<void*>(lib_handle->op_list.data)); in TF_DeleteLibraryHandle()570 delete lib_handle; in TF_DeleteLibraryHandle()
66 void* lib_handle; member
330 TF_Library* lib_handle);
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);
283 loader_platform_dl_handle lib_handle = LoadLibrary(lib_path); in loader_platform_open_library() local284 if (lib_handle == NULL && GetLastError() == ERROR_MOD_NOT_FOUND) { in loader_platform_open_library()286 …lib_handle = LoadLibraryEx(lib_path, NULL, LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LOAD_LIBRARY_SEARCH_… in loader_platform_open_library()288 return lib_handle; in loader_platform_open_library()
752 m.def("TF_GetOpList", [](TF_Library* lib_handle) { in PYBIND11_MODULE() argument753 TF_Buffer output_buffer = TF_GetOpList(lib_handle); in PYBIND11_MODULE()