Home
last modified time | relevance | path

Searched refs:mark_variable_ready_dense (Results 1 – 2 of 2) sorted by relevance

/external/pytorch/torch/csrc/distributed/c10d/
Dreducer.hpp272 void mark_variable_ready_dense(size_t variable_index);
Dreducer.cpp348 void Reducer::mark_variable_ready_dense(size_t variable_index) { in mark_variable_ready_dense() function in c10d::Reducer
593 mark_variable_ready_dense(variable_index); in delay_all_reduce()
895 mark_variable_ready_dense(variable_index); in mark_variable_ready()