Searched defs:JaxPmapFunctionObject (Results 1 – 1 of 1) sorted by relevance
565 struct JaxPmapFunctionObject { struct567 PyObject* dict; // Dictionary for __dict__568 PyObject* weakrefs; // Weak references; for use by the Python interpreter.569 PmapFunction fun;