(self)
| 69 | |
| 70 | @gpu_test |
| 71 | def test_gpu(self): |
| 72 | with tf.device('/gpu:0'): |
| 73 | m1 = tf.constant([2.0, 3.0], shape=[1, 2], name='a') |
| 74 | m2 = tf.constant([3.0, 4.0], shape=[2, 1], name='b') |
| 75 | result = tf.matmul(m1, m2) |
| 76 | self.assertEqual([1, 1], result.shape) |
| 77 | |
| 78 | @gpu_test |
| 79 | def test_is_built_with_cuda(self): |
nothing calls this directly
no outgoing calls
no test coverage detected