Searched refs:py_class (Results 1 – 6 of 6) sorted by relevance
46 bool IsRegisteredDispatchableType(PyObject* py_class) { in IsRegisteredDispatchableType() argument49 int result = PyObject_IsSubclass(py_class, registered_type.get()); in IsRegisteredDispatchableType()74 bool RegisterDispatchableType(PyObject* py_class) { in RegisterDispatchableType() argument76 if (!PyType_Check(py_class)) { in RegisterDispatchableType()80 py_class->ob_type->tp_name) in RegisterDispatchableType()84 if (IsRegisteredDispatchableType(py_class)) { in RegisterDispatchableType()85 Safe_PyObjectPtr s(PyObject_Str(py_class)); in RegisterDispatchableType()93 Py_INCREF(py_class); in RegisterDispatchableType()94 GetRegisteredDispatchableTypes().push_back(Safe_PyObjectPtr(py_class)); in RegisterDispatchableType()212 for (PyObject* py_class : py_classes) { in PyInstanceChecker()[all …]
138 m.def("register_dispatchable_type", [](py::handle py_class) { in PYBIND11_MODULE() argument139 if (!tensorflow::py_dispatch::RegisterDispatchableType(py_class.ptr())) { in PYBIND11_MODULE()142 return py_class; in PYBIND11_MODULE()
126 bool RegisterDispatchableType(PyObject* py_class);
835 auto py_class = in getScriptedClassOrError() local838 if (py_class.is_none()) { in getScriptedClassOrError()845 return py_class; in getScriptedClassOrError()
725 auto py_class = getScriptedClassOrError(enum_holder->type()); in toPyObject() local726 return py_class.attr(enum_holder->name().c_str()); in toPyObject()
1049 orig_class = concrete_type.py_class