Searched defs:WorkInfo (Results 1 – 2 of 2) sorted by relevance
145 struct TORCH_API WorkInfo { struct146 WorkInfo( in WorkInfo() function158 OpType opType;159 uint64_t seq;160 std::chrono::time_point<std::chrono::system_clock> timeStarted;161 std::chrono::time_point<std::chrono::system_clock> timeFinished;162 std::chrono::duration<float> activeDuration;
2248 def hook(work_info: torch._C._distributed_c10d.WorkInfo):2276 def hook(work_info: torch._C._distributed_c10d.WorkInfo):2318 def hook(work_info: torch._C._distributed_c10d.WorkInfo):2376 def hook(work_info: torch._C._distributed_c10d.WorkInfo):2414 def hook(work_info: torch._C._distributed_c10d.WorkInfo):