Searched refs:num_interop_threads (Results 1 – 1 of 1) sorted by relevance
20 std::atomic<int> num_interop_threads{NOT_SET}; variable29 /* pool_size */ num_interop_threads.exchange(CONSUMED), in get_pool()54 TORCH_CHECK(num_interop_threads.compare_exchange_strong(no_value, nthreads), in set_num_interop_threads()61 int nthreads = num_interop_threads.load(); in get_num_interop_threads()