Method test_is_built_with_cuda

tests/test_tensorflow.py:79–80

(self)

    def test_gpu(self):
        with tf.device('/gpu:0'):
            m1 = tf.constant([2.0, 3.0], shape=[1, 2], name='a')
            m2 = tf.constant([3.0, 4.0], shape=[2, 1], name='b')
            result = tf.matmul(m1, m2)
            self.assertEqual([1, 1], result.shape)