Function tch::utils::set_num_interop_threads
source · pub fn set_num_interop_threads(n_threads: i32)
Expand description
Set the number of threads used by torch for inter-op parallelism.
pub fn set_num_interop_threads(n_threads: i32)
Set the number of threads used by torch for inter-op parallelism.