Method image

tensorboard/plugins/image/summary_test.py:159–165

(self, *args, **kwargs)

class SummaryV1OpTest(SummaryBaseTest, tf.test.TestCase):
    def image(self, *args, **kwargs):
        args = list(args)
        # Force first argument to tf.uint8 since the V1 version requires this.
        args[1] = tf.cast(tf.constant(args[1]), tf.uint8)
        return summary_pb2.Summary.FromString(
            summary.op(*args, **kwargs).numpy()
        )

    def test_tag(self):
        data = np.array(1, np.uint8, ndmin=4)