Searched refs:numGradHooksTriggeredMapPerIteration_ (Results 1 – 2 of 2) sorted by relevance
708 numGradHooksTriggeredMapPerIteration_[index] > 0, in autograd_hook()714 if (--numGradHooksTriggeredMapPerIteration_[index] == 0) { in autograd_hook()1335 numGradHooksTriggeredMapPerIteration_ = numGradHooksTriggeredMap_; in reset_bucket_counting()
503 std::unordered_map<size_t, int> numGradHooksTriggeredMapPerIteration_; member in c10d::Reducer