Function tch::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.