Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp723 PyObject* THPModule_setSDPUseMath(PyObject* _unused, PyObject* arg) { in THPModule_setSDPUseMath() function
1384 {"_set_sdp_use_math", THPModule_setSDPUseMath, METH_O, nullptr},
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1144 def _set_sdp_use_math(arg: _bool) -> None: ... # THPModule_setSDPUseMath
1146 def _set_sdp_use_cudnn(arg: _bool) -> None: ... # THPModule_setSDPUseMath