Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/profiler/
H A Dcollection.h675 TORCH_API void set_cuda_sync_enabled_fn(std::function<bool()>);
H A Dcollection.cpp1573 void set_cuda_sync_enabled_fn(std::function<bool()> fn) { in set_cuda_sync_enabled_fn() function