Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A Dreducer.hpp445 void record_forward_compute_start_time();
H A Dreducer.cpp1320 record_forward_compute_start_time(); in prepare_for_forward()
2024 void Reducer::record_forward_compute_start_time() { in record_forward_compute_start_time() function in c10d::Reducer