Searched refs:PythonAPIDispatcher (Results 1 – 6 of 6) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | python_api_dispatcher_wrapper.cc | 28 using tensorflow::py_dispatch::PythonAPIDispatcher;
34 py::object Dispatch(PythonAPIDispatcher* self, py::handle args, in Dispatch()
44 PythonAPIDispatcher MakePythonAPIDispatcher( in MakePythonAPIDispatcher()
54 return PythonAPIDispatcher(api_name, arg_names_span, {}); in MakePythonAPIDispatcher()
61 return PythonAPIDispatcher( in MakePythonAPIDispatcher()
103 py::class_<PythonAPIDispatcher>(m, "PythonAPIDispatcher") in PYBIND11_MODULE()
106 [](PythonAPIDispatcher* self, PySignatureChecker signature_checker, in PYBIND11_MODULE()
112 [](PythonAPIDispatcher* self, py::handle func) { in PYBIND11_MODULE()
115 .def("__repr__", &PythonAPIDispatcher::DebugString); in PYBIND11_MODULE()
|
| H A D | python_api_dispatcher.cc | 98 PythonAPIDispatcher::PythonAPIDispatcher(const std::string& api_name, in PythonAPIDispatcher() function in tensorflow::py_dispatch::PythonAPIDispatcher
106 void PythonAPIDispatcher::Register(PySignatureChecker signature_checker, in Register()
114 Safe_PyObjectPtr PythonAPIDispatcher::Dispatch(PyObject* args, in Dispatch()
144 void PythonAPIDispatcher::Unregister(PyObject* func) { in Unregister()
154 std::string PythonAPIDispatcher::DebugString() const { in DebugString()
|
| H A D | python_api_dispatcher.h | 61 class PythonAPIDispatcher {
71 PythonAPIDispatcher(const std::string& api_name,
|
| H A D | python_api_dispatcher_test.py | 334 dispatcher = dispatch.PythonAPIDispatcher('tf.foo', ['x', 'y', 'name'],
357 dispatcher = dispatch.PythonAPIDispatcher('tf.foo', ['x', 'y', 'name'],
382 dispatcher = dispatch.PythonAPIDispatcher('tf.foo', ['x', 'ys', 'name'],
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/util/ |
| H A D | dispatch.py | 562 _api_dispatcher.PythonAPIDispatcher(unwrapped.__name__,
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/def_file_filter/ |
| H A D | symbols_pybind.txt | 454 tensorflow::py_dispatch::PythonAPIDispatcher
|