Function tch::set_num_interop_threads[][src]

pub fn set_num_interop_threads(n_threads: i32)

Set the number of threads used by torch for inter-op parallelism.