Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp643 PyObject* THPModule_setAllowTF32CuDNN(PyObject* _unused, PyObject* arg) { in THPModule_setAllowTF32CuDNN() function
1411 {"_set_cudnn_allow_tf32", THPModule_setAllowTF32CuDNN, METH_O, nullptr},
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1167 def _set_cudnn_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32CuDNN