Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/core/dispatch/
H A DDispatcher.cpp277 PythonModuleMapType& pythonModulesSingleton() { in pythonModulesSingleton() function
286 auto found = pythonModulesSingleton().find(op_name); in getPyStub()
287 if (found == pythonModulesSingleton().end()) { in getPyStub()
302 auto found = pythonModulesSingleton().find(op_name); in registerPythonModule()
303 if (found != pythonModulesSingleton().end()) { in registerPythonModule()
310 pythonModulesSingleton()[op_name] = std::make_pair(pymodule, context); in registerPythonModule()
316 pythonModulesSingleton().erase(op_name); in registerPythonModule()
322 auto elt = pythonModulesSingleton().find(op_name); in throwIfHasPythonModule()
323 if (elt == pythonModulesSingleton().end()) { in throwIfHasPythonModule()