Searched refs:THPModule_setAllowTF32CuDNN (Results 1 – 2 of 2) sorted by relevance
643 PyObject* THPModule_setAllowTF32CuDNN(PyObject* _unused, PyObject* arg) { in THPModule_setAllowTF32CuDNN() function1411 {"_set_cudnn_allow_tf32", THPModule_setAllowTF32CuDNN, METH_O, nullptr},
1167 def _set_cudnn_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32CuDNN