deepmd.tf.env
Module that sets tensorflow working environment and exports inportant constants. Limit tensorflow session to CPU or not. Cast tensor to globally set TF precision. Cast tensor to globally set energy precision. Limit tensorflow session to CPU or not. If enabled, no GPU device is visible to the TensorFlow Session.Module Contents
Functions
reset_default_tf_session_config
(cpu_only)global_cvt_2_tf_float
(→ tensorflow.compat.v1.Tensor)global_cvt_2_ener_float
(→ tensorflow.compat.v1.Tensor)Attributes