Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp275 bool Reducer::static_graph_first_iteration() { in static_graph_first_iteration() function in c10d::Reducer
422 this->static_graph_first_iteration()) { in mark_variable_ready_dense()
651 if (dynamic_graph_find_unused() || static_graph_first_iteration()) { in autograd_hook()
670 if (static_graph_first_iteration()) { in autograd_hook()
1664 if (dynamic_graph_find_unused() || static_graph_first_iteration()) { in finalize_backward()
H A Dreducer.hpp522 bool static_graph_first_iteration();