(self)
| 583 | self.assertEqual("/job:goo/device:GPU:0", op.device) |
| 584 | |
| 585 | def testDeviceObject(self): |
| 586 | op = ops.Operation(ops._NodeDef("None", "myop"), ops.Graph(), [], []) |
| 587 | op._set_device("/job:goo/device:GPU:0") |
| 588 | self.assertProtoEquals( |
| 589 | "op:'None' name:'myop' device:'/job:goo/device:GPU:0' ", op.node_def) |
| 590 | op = ops.Operation(ops._NodeDef("None", "op2"), ops.Graph(), [], []) |
| 591 | op._set_device( |
| 592 | pydev.DeviceSpec( |
| 593 | job="muu", device_type="CPU", device_index=0)) |
| 594 | self.assertProtoEquals( |
| 595 | "op:'None' name:'op2' device:'/job:muu/device:CPU:0'", op.node_def) |
| 596 | |
| 597 | def testReferenceInput(self): |
| 598 | g = ops.Graph() |
nothing calls this directly
no test coverage detected