def __add__(self, other):
    """Add to another LazyAllreduceSum.

    Args:
      other: a LazyAllreduceSum or a LaidOutTensor
    Returns:
      a LazyAllreduceSum or a LaidOutTensor
    """
    if (isinstance(other, LazyAllreduceSum) and
        self.mesh_impl == other.mesh_impl and
        self.mesh_axes == other.mesh_axes):
      return LazyAllreduceSum(
          self.mesh_impl,
          self.mesh_impl.slicewise(
              tf.add, self.laid_out_input, other.laid_out_input),
          self.mesh_axes,
          add_counter_fn=self.add_counter_fn)
    else:
      return self.mesh_impl.slicewise(
          tf.add, self.to_laid_out_tensor(), other.to_laid_out_tensor())
def __add__ ( self , other ) : if ( isinstance ( other , LazyAllreduceSum ) and self . mesh_impl == other . mesh_impl and self . mesh_axes == other . mesh_axes ) : return LazyAllreduceSum ( self . mesh_impl , self . mesh_impl . slicewise ( tf . add , self . laid_out_input , other . laid_out_input ) , self . mesh_axes , add_counter_fn = self . add_counter_fn ) else : return self . mesh_impl . slicewise ( tf . add , self . to_laid_out_tensor ( ) , other . to_laid_out_tensor ( ) )
Add to another LazyAllreduceSum.

    Args:
      other: a LazyAllreduceSum or a LaidOutTensor
    Returns:
      a LazyAllreduceSum or a LaidOutTensor