Home
last modified time | relevance | path

Searched refs:EmitPythonCallback (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/python/
H A Dpy_client.h203 StatusOr<std::pair<XlaOp, pybind11::object>> EmitPythonCallback(
H A Dxla.cc263 .def("emit_python_callback", &PyClient::EmitPythonCallback, in PYBIND11_MODULE()
H A Dpy_client.cc688 StatusOr<std::pair<XlaOp, pybind11::object>> PyClient::EmitPythonCallback( in EmitPythonCallback() function in xla::PyClient