Ejemplo n.º 1
0
    def compute_and_check_summary_pb(self,
                                     name,
                                     data,
                                     display_name=None,
                                     description=None,
                                     data_tensor=None,
                                     feed_dict=None):
        """Use both `op` and `pb` to get a summary, asserting equality.

    Returns:
      a `Summary` protocol buffer
    """
        if data_tensor is None:
            data_tensor = tf.constant(data)
        op = summary.op(name,
                        data,
                        display_name=display_name,
                        description=description)
        pb = self.normalize_summary_pb(
            summary.pb(name,
                       data,
                       display_name=display_name,
                       description=description))
        pb_via_op = self.normalize_summary_pb(
            self.pb_via_op(op, feed_dict=feed_dict))
        self.assertProtoEquals(pb, pb_via_op)
        return pb
Ejemplo n.º 2
0
  def compute_and_check_summary_pb(self, name, data,
                                   display_name=None, description=None,
                                   data_tensor=None, feed_dict=None):
    """Use both `op` and `pb` to get a summary, asserting equality.

    Returns:
      a `Summary` protocol buffer
    """
    if data_tensor is None:
      data_tensor = tf.constant(data)
    op = summary.op(
        name, data, display_name=display_name, description=description)
    pb = self.normalize_summary_pb(summary.pb(
        name, data, display_name=display_name, description=description))
    pb_via_op = self.normalize_summary_pb(
        self.pb_via_op(op, feed_dict=feed_dict))
    self.assertProtoEquals(pb, pb_via_op)
    return pb
Ejemplo n.º 3
0
 def text(self, *args, **kwargs):
     return summary.pb(*args, **kwargs)
Ejemplo n.º 4
0
 def test_non_string_value_in_pb(self):
   with six.assertRaisesRegex(
       self,
       ValueError,
       r'Expected binary or unicode string, got 0'):
     summary.pb('la', np.array(range(42)))