Searched refs:set_num_interop_threads (Results 1 – 14 of 14) sorted by relevance
39 at::set_num_interop_threads(5); in main()41 ASSERT_ANY_THROW(at::set_num_interop_threads(6)); in main()
11 torch::set_num_interop_threads(2); in TEST()
89 using at::set_num_interop_threads;
50 void set_num_interop_threads(int nthreads) { in set_num_interop_threads() function
133 TORCH_API void set_num_interop_threads(int);
62 at::set_num_interop_threads(FLAGS_inter_op_threads); in main()
96 | Inter-op parallelism | ``at::set_num_interop_threads``, | Default numb…99 | | ``set_num_interop_threads``, | …
293 torch.set_num_interop_threads(args.num_threads)
256 set_num_interop_threads
9576 torch.set_num_interop_threads,
148 torch.set_num_interop_threads,
342 at::set_num_interop_threads(nthreads); in THPModule_setNumInteropThreads()
1132 def set_num_interop_threads(
10560 self.assertRaises(RuntimeError, lambda: torch.set_num_interop_threads(invalid_val))