Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.hpp446 void record_backward_compute_start_time();
H A Dreducer.cpp1431 record_backward_compute_start_time(); in prepare_for_backward()
2030 void Reducer::record_backward_compute_start_time() { in record_backward_compute_start_time() function in c10d::Reducer