Searched defs:set_autocast_enabled (Results 1 – 3 of 3) sorted by relevance
14 void set_autocast_enabled(at::DeviceType device_type, bool enabled) { in set_autocast_enabled() function
1024 def set_autocast_enabled(enable): member in DummyPrivateUse1Module
536 static PyObject* set_autocast_enabled( in set_autocast_enabled() function