Searched refs:THPModule_setSDPUseMath (Results 1 – 2 of 2) sorted by relevance
723 PyObject* THPModule_setSDPUseMath(PyObject* _unused, PyObject* arg) { in THPModule_setSDPUseMath() function1384 {"_set_sdp_use_math", THPModule_setSDPUseMath, METH_O, nullptr},
1144 def _set_sdp_use_math(arg: _bool) -> None: ... # THPModule_setSDPUseMath1146 def _set_sdp_use_cudnn(arg: _bool) -> None: ... # THPModule_setSDPUseMath