(self)
| 381 | del ctx |
| 382 | |
| 383 | def testMultiCpuPlacement(self): |
| 384 | with ops.device('cpu:1'): |
| 385 | x = constant_op.constant(1.0) |
| 386 | y = array_ops.identity(x) |
| 387 | self.assertEqual(x.device, '/job:localhost/replica:0/task:0/device:CPU:1') |
| 388 | self.assertEqual(y.device, '/job:localhost/replica:0/task:0/device:CPU:0') |
| 389 | |
| 390 | @test_util.run_gpu_only |
| 391 | def testShouldCopy(self): |