Commit 185d391d authored by Bharath Ramsundar's avatar Bharath Ramsundar
Browse files

Merge branch 'master' of https://github.com/deepchem/deepchem into tg_api

parents ae54ef49 614d8a3c

deepchem/models/tensorgraph/layers.py

+3 −0
Original line number Diff line number Diff line
@@ -162,7 +162,10 @@ class Layer(object): 
      tf.summary.histogram(self.name, self.tb_input, self.collections)



  def _as_graph_element(self):

    if '_as_graph_element' in dir(self.out_tensor):

      return self.out_tensor._as_graph_element()

    else:

      return self.out_tensor





def _convert_layer_to_tensor(value, dtype=None, name=None, as_ref=False):

CRA Git | Maintained and supported by SUSTech CRA and CCSE