Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/
H A DModule.cpp1001 PyObject* THPModule_setAllowTF32CuBLAS(PyObject* _unused, PyObject* arg) { in THPModule_setAllowTF32CuBLAS() function
1457 {"_set_cublas_allow_tf32", THPModule_setAllowTF32CuBLAS, METH_O, nullptr},
/aosp_15_r20/external/pytorch/torch/_C/
H A D__init__.pyi.in1169 def _set_cublas_allow_tf32(arg: _bool) -> None: ... # THPModule_setAllowTF32CuBLAS