(self, DeviceSpec)
| 37 | |
| 38 | @parameterized.named_parameters(*TEST_V1_AND_V2) |
| 39 | def testMerge(self, DeviceSpec): # pylint: disable=invalid-name |
| 40 | d = DeviceSpec.from_string("/job:muu/task:1/device:MyFunnyDevice:2") |
| 41 | self.assertEqual("/job:muu/task:1/device:MyFunnyDevice:2", d.to_string()) |
| 42 | |
| 43 | if not context.executing_eagerly(): |
| 44 | with ops.device(device.merge_device("/device:GPU:0")): |
| 45 | var1 = variables.Variable(1.0) |
| 46 | self.assertEqual("/device:GPU:0", var1.device) |
| 47 | with ops.device(device.merge_device("/job:worker")): |
| 48 | var2 = variables.Variable(1.0) |
| 49 | self.assertEqual("/job:worker/device:GPU:0", var2.device) |
| 50 | with ops.device(device.merge_device("/device:CPU:0")): |
| 51 | var3 = variables.Variable(1.0) |
| 52 | self.assertEqual("/job:worker/device:CPU:0", var3.device) |
| 53 | with ops.device(device.merge_device("/job:ps")): |
| 54 | var4 = variables.Variable(1.0) |
| 55 | self.assertEqual("/job:ps/device:CPU:0", var4.device) |
| 56 | |
| 57 | def testCanonicalName(self): |
| 58 | self.assertEqual("/job:foo/replica:0", |
nothing calls this directly
no test coverage detected