code: def export_to_tf_tensor(self, x):
      
      mesh_impl = self.mesh_impl(x)
      return mesh_impl.export_to_tf_tensor(
          x, self.tensors[x].to_laid_out_tensor())
docs: Turn a Tensor into a tf.Tensor.
  
      Args:
        x: Tensor.
  
      Returns:
        tf.Tensor.