deepmd.tf.utils.compress

Attributes

log

Functions

get_type_embedding(self, graph)
get_two_side_type_embedding(self, graph)
get_extra_side_embedding_net_variable(self, graph_def, ...)
_layer_0(self, x, w, b)
_layer_1(self, x, w, b)
make_data(self, xx)

Module Contents

deepmd.tf.utils.compress.log

deepmd.tf.utils.compress.get_type_embedding(self, graph)

deepmd.tf.utils.compress.get_two_side_type_embedding(self, graph)

deepmd.tf.utils.compress.get_extra_side_embedding_net_variable(self, graph_def, type_side_suffix, varialbe_name, suffix)

deepmd.tf.utils.compress._layer_0(self, x, w, b)

deepmd.tf.utils.compress._layer_1(self, x, w, b)

deepmd.tf.utils.compress.make_data(self, xx)