Searched refs:mark_variable_ready (Results 1 – 2 of 2) sorted by relevance
692 mark_variable_ready(unused_index); in autograd_hook()719 mark_variable_ready(index); in autograd_hook()726 mark_variable_ready(index); in autograd_hook()870 void Reducer::mark_variable_ready(size_t variable_index) { in mark_variable_ready() function in c10d::Reducer
276 void mark_variable_ready(size_t variable_index);