message: Adding tf.name_scope blocks to make the TensorBoard graph visualization usable.
diff: @@ -265,10 +265,13 @@ class WALSModel(object):
  "col_factors")
  self._row_gramian = self._create_gramian(self._n_components, "row_gramian")
  self._col_gramian = self._create_gramian(self._n_components, "col_gramian")
  + with ops.name_scope("row_prepare_gramian"):
  self._row_update_prep_gramian = self._prepare_gramian(
  self._col_factors, self._col_gramian)
  + with ops.name_scope("col_prepare_gramian"):
  self._col_update_prep_gramian = self._prepare_gramian(
  self._row_factors, self._row_gramian)
  + with ops.name_scope("transient_vars"):
  self._create_transient_vars()
  @property
  @@ -310,6 +313,7 @@ class WALSModel(object):
  @classmethod
  def _create_factors(cls, rows, cols, num_shards, init, name):
  """Helper function to create row and column factors."""
  + with ops.name_scope(name):
  if callable(init):
  init = init()
  if isinstance(init, list):
  @@ -380,6 +384,7 @@ class WALSModel(object):
  sizes = cls._shard_sizes(num_wts, num_shards)
  assert len(sizes) == num_shards
  + with ops.name_scope(name):
  def make_wt_initializer(i, size):
  def initializer():
  