Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp708 numGradHooksTriggeredMapPerIteration_[index] > 0, in autograd_hook()
714 if (--numGradHooksTriggeredMapPerIteration_[index] == 0) { in autograd_hook()
1335 numGradHooksTriggeredMapPerIteration_ = numGradHooksTriggeredMap_; in reset_bucket_counting()
H A Dreducer.hpp503 std::unordered_map<size_t, int> numGradHooksTriggeredMapPerIteration_; member in c10d::Reducer