Function deepmd::get_env_nthreads¶
Defined in File common.h
Function Documentation¶
-
void deepmd::get_env_nthreads(int &num_intra_nthreads, int &num_inter_nthreads)¶
Get the number of threads from the environment variable.
- Parameters
num_intra_nthreads – [out] The number of intra threads. Read from TF_INTRA_OP_PARALLELISM_THREADS.
num_inter_nthreads – [out] The number of inter threads. Read from TF_INTER_OP_PARALLELISM_THREADS.