Searched refs:EagerContextThreadLocalDataWrapper (Results 1 – 1 of 1) sorted by relevance
376 class EagerContextThreadLocalDataWrapper { class378 explicit EagerContextThreadLocalDataWrapper(py::handle py_eager_context, in EagerContextThreadLocalDataWrapper() function in __anonaa2eb7e00311::EagerContextThreadLocalDataWrapper386 ~EagerContextThreadLocalDataWrapper() { in ~EagerContextThreadLocalDataWrapper()1458 py::class_<EagerContextThreadLocalDataWrapper>(m, in PYBIND11_MODULE()1464 &EagerContextThreadLocalDataWrapper::get_is_eager, in PYBIND11_MODULE()1465 &EagerContextThreadLocalDataWrapper::set_is_eager) in PYBIND11_MODULE()1468 &EagerContextThreadLocalDataWrapper::get_invoking_op_callbacks, in PYBIND11_MODULE()1469 &EagerContextThreadLocalDataWrapper::set_invoking_op_callbacks) in PYBIND11_MODULE()1471 &EagerContextThreadLocalDataWrapper::get_device_name, in PYBIND11_MODULE()1472 &EagerContextThreadLocalDataWrapper::set_device_name) in PYBIND11_MODULE()[all …]