Searched refs:set_autocast_cpu_enabled (Results 1 – 3 of 3) sorted by relevance
647 static PyObject* set_autocast_cpu_enabled(PyObject* _unused, PyObject* arg) { in set_autocast_cpu_enabled() function1317 {"set_autocast_cpu_enabled", set_autocast_cpu_enabled, METH_O, nullptr},
274 torch.set_autocast_cpu_enabled,
1304 def set_autocast_cpu_enabled(enabled: _bool) -> None: ...