abaa902d9c27646a0528ab7c8b6ee182dce1de2c,deepchem/models/tensorgraph/layers.py,BatchNorm,create_tensor,#BatchNorm#Any#Any#,3140

Before Change

  def create_tensor(self, in_layers=None, set_tensors=True, **kwargs):
    inputs = self._get_input_tensors(in_layers)
    parent_tensor = inputs[0]
    out_tensor = tf.layers.batch_normalization(parent_tensor)
    if set_tensors:
      self.out_tensor = out_tensor
    return out_tensor

After Change

    inputs = self._get_input_tensors(in_layers)
    parent_tensor = inputs[0]
    if tfe.in_eager_mode():
      if not self._built:
        self._layer = self._build_layer()
      layer = self._layer
    else:
      layer = self._build_layer()
    out_tensor = layer(parent_tensor)

Instances