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
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
def text(self, *args, **kwargs): return summary.pb(*args, **kwargs)
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)))