Searched refs:assertDeviceEqual (Results 1 – 9 of 9) sorted by relevance
42 self.assertDeviceEqual("/job:ps/task:0/cpu:0", v.device)43 self.assertDeviceEqual("/job:ps/task:0/cpu:0", v.initializer.device)44 self.assertDeviceEqual("/job:ps/task:1", w.device)45 self.assertDeviceEqual("/job:ps/task:1", w.initializer.device)46 self.assertDeviceEqual("/job:worker/cpu:0", a.device)53 self.assertDeviceEqual("/job:ps/task:0", v.device)62 self.assertDeviceEqual("/job:ps/task:0", v.device)63 self.assertDeviceEqual("/job:ps/task:0", v.initializer.device)64 self.assertDeviceEqual("/job:ps/task:1", w.device)65 self.assertDeviceEqual("/job:ps/task:1", w.initializer.device)[all …]
544 self.assertDeviceEqual("/job:dev_v0", ema.average(v0).device)545 self.assertDeviceEqual("/job:dev_v1", ema.average(v1).device)548 self.assertDeviceEqual("/job:default", ema.average(tensor2).device)
91 self.assertDeviceEqual(var_0.device, None)92 self.assertDeviceEqual(var_1.device, "/device:CPU:0")93 self.assertDeviceEqual(var_2.device, None)94 self.assertDeviceEqual(var_3.device, None)95 self.assertDeviceEqual(var_4.device, "/device:CPU:0")96 self.assertDeviceEqual(var_5.device, "/device:GPU:0")97 self.assertDeviceEqual(var_6.device, "/device:CPU:0")110 self.assertDeviceEqual(var_0.device, None)111 self.assertDeviceEqual(var_1.device, "/device:CPU:0")112 self.assertDeviceEqual(var_2.device, "/device:GPU:0")[all …]
1187 self.assertDeviceEqual(None, op1.device)1188 self.assertDeviceEqual("/device:GPU:0", op2.device)1189 self.assertDeviceEqual(None, op3.device)1664 self.assertDeviceEqual(None, op.device)
1329 self.assertDeviceEqual(node.device, "/job:ADevice")
3497 def assertDeviceEqual(self, device1, device2, msg=None): member in TensorFlowTestCase
776 self.assertDeviceEqual(None, var.device)777 self.assertDeviceEqual(None, var.initializer.device)783 self.assertDeviceEqual("/job:ps", var.device)784 self.assertDeviceEqual("/job:ps", var.initializer.device)
461 self.assertDeviceEqual("/job:ps", assign_op.device)
574 self.assertDeviceEqual(op.device, "/cpu:0")4421 self.assertDeviceEqual(None, with_vnod_dep.device)4429 self.assertDeviceEqual("", with_vdef_dep.device)