Searched defs:prev_grad_mode (Results 1 – 4 of 4) sorted by relevance
32 std::optional<bool> prev_grad_mode, in DynamicLayer()255 std::optional<bool> prev_grad_mode, in initAndPushDynamicLayer()
87 std::optional<bool> prev_grad_mode, in autogradBasedTransformSendToNext()
184 def prev_grad_mode(self): member in GradInterpreter
243 bool prev_grad_mode = c10::GradMode::is_enabled(); in _grad_increment_nesting() local