(self)
| 483 | |
| 484 | @test_util.run_gpu_only |
| 485 | def testReEntrant(self): |
| 486 | cpu = context.device('cpu:0') |
| 487 | gpu = context.device('gpu:0') |
| 488 | with cpu: |
| 489 | with gpu: |
| 490 | with gpu: |
| 491 | self.assertEndsWith(current_device(), 'GPU:0') |
| 492 | self.assertEndsWith(current_device(), 'GPU:0') |
| 493 | self.assertEndsWith(current_device(), 'CPU:0') |
| 494 | with gpu: |
| 495 | self.assertEndsWith(current_device(), 'GPU:0') |
| 496 | |
| 497 | @test_util.run_gpu_only |
| 498 | def testTensorPlacement(self): |
nothing calls this directly
no test coverage detected