Searched defs:AutogradState (Results 1 – 1 of 1) sorted by relevance
9 struct C10_API AutogradState { struct13 AutogradState( in AutogradState() argument24 void set_grad_mode(bool enabled) { in set_grad_mode()28 void set_fw_grad_mode(bool enabled) { in set_fw_grad_mode()32 void set_inference_mode(bool enabled) { in set_inference_mode()36 void set_multithreading_enabled(bool multithreading_enabled) { in set_multithreading_enabled()40 void set_view_replay_enabled(bool view_replay_enabled) { in set_view_replay_enabled()44 bool get_grad_mode() const { in get_grad_mode()48 bool get_fw_grad_mode() const { in get_fw_grad_mode()52 bool get_inference_mode() const { in get_inference_mode()[all …]