Searched refs:THPModule_setAllowTF32CuBLAS (Results 1 – 2 of 2) sorted by relevance
1001 PyObject* THPModule_setAllowTF32CuBLAS(PyObject* _unused, PyObject* arg) { in THPModule_setAllowTF32CuBLAS() function1457 {"_set_cublas_allow_tf32", THPModule_setAllowTF32CuBLAS, METH_O, nullptr},
1169 def _set_cublas_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32CuBLAS