Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/util/
H A Dfast_module_type_test.py19 FastModuleType = fast_module_type.get_fast_module_type_class() variable
22 class ChildFastModule(FastModuleType):
45 FastModuleType.set_getattribute_callback(module,
52 FastModuleType.set_getattribute_callback(module,
54 FastModuleType.set_getattr_callback(module, ChildFastModule._getattr)
H A Dfast_module_type.cc233 static PyTypeObject FastModuleType = []() { in __anon529087110102() variable
263 DCHECK(IsAnyBaseSameType(obj, &FastModuleType)); in UncheckedCast()
268 FastModuleType.tp_base = &PyModule_Type; in PYBIND11_MODULE()
269 FastModuleType.tp_setattro = [](PyObject *module, PyObject *name, in PYBIND11_MODULE()
294 reinterpret_cast<PyObject *>(&FastModuleType)); in PYBIND11_MODULE()
H A Dmodule_wrapper.py26 FastModuleType = fast_module_type.get_fast_module_type_class() variable
90 class TFModuleWrapper(FastModuleType):
103 FastModuleType.set_getattr_callback(self, TFModuleWrapper._getattr)
104 FastModuleType.set_getattribute_callback(self,