Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.cpp571 record_backward_compute_end_time(); in delay_all_reduce()
919 record_backward_compute_end_time(); in mark_variable_ready()
2036 void Reducer::record_backward_compute_end_time() { in record_backward_compute_end_time() function in c10d::Reducer
H A Dreducer.hpp447 void record_backward_compute_end_time();