Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.hpp240 std::vector<size_t> prev_iteration_unused_parameters_; member in c10d::Reducer
H A Dreducer.cpp1413 prev_iteration_unused_parameters_ == unused_parameters_; in search_unused_parameters()
1421 prev_iteration_unused_parameters_ = unused_parameters_; in search_unused_parameters()