Searched refs:set_optimizer_in_backward (Results 1 – 2 of 2) sorted by relevance
104 void set_optimizer_in_backward() { in set_optimizer_in_backward() function in c10d::Reducer
603 [](::c10d::Reducer& reducer) { reducer.set_optimizer_in_backward(); }, in c10d_init()