Searched refs:num_threads_in_backwards (Results 1 – 1 of 1) sorted by relevance
79 std::atomic<int32_t> num_threads_in_backwards; variable82 num_threads_in_backwards++; in CompiledAutogradThreadingDebugCheck()89 num_threads_in_backwards--; in release()1405 num_threads_in_backwards.load() == 0 && prior != COMPILED_AUTOGRAD_POISON, in set_compiled_autograd()