Searched refs:set_cuda_sync_enabled_fn (Results 1 – 2 of 2) sorted by relevance
675 TORCH_API void set_cuda_sync_enabled_fn(std::function<bool()>);
1573 void set_cuda_sync_enabled_fn(std::function<bool()> fn) { in set_cuda_sync_enabled_fn() function