Searched refs:remove_autograd_hooks (Results 1 – 3 of 3) sorted by relevance
186 void remove_autograd_hooks();
268 remove_autograd_hooks(); in ~Reducer()2332 void Reducer::remove_autograd_hooks() { in remove_autograd_hooks() function in c10d::Reducer
677 [](::c10d::Reducer& reducer) { reducer.remove_autograd_hooks(); }, in c10d_init()