Searched refs:assertDeviceEqual (Results 1 – 9 of 9) sorted by relevance
46 self.assertDeviceEqual("/job:ps/task:0/cpu:0", v.device)47 self.assertDeviceEqual("/job:ps/task:0/cpu:0", v.initializer.device)48 self.assertDeviceEqual("/job:ps/task:1", w.device)49 self.assertDeviceEqual("/job:ps/task:1", w.initializer.device)50 self.assertDeviceEqual("/job:worker/cpu:0", a.device)57 self.assertDeviceEqual("/job:ps/task:0", v.device)66 self.assertDeviceEqual("/job:ps/task:0", v.device)67 self.assertDeviceEqual("/job:ps/task:0", v.initializer.device)68 self.assertDeviceEqual("/job:ps/task:1", w.device)69 self.assertDeviceEqual("/job:ps/task:1", w.initializer.device)[all …]
464 self.assertDeviceEqual("/job:dev_v0", ema.average(v0).device)465 self.assertDeviceEqual("/job:dev_v1", ema.average(v1).device)468 self.assertDeviceEqual("/job:default", ema.average(tensor2).device)
95 self.assertDeviceEqual(var_0.device, None)96 self.assertDeviceEqual(var_1.device, "/device:CPU:0")97 self.assertDeviceEqual(var_2.device, None)98 self.assertDeviceEqual(var_3.device, None)99 self.assertDeviceEqual(var_4.device, "/device:CPU:0")100 self.assertDeviceEqual(var_5.device, "/device:GPU:0")101 self.assertDeviceEqual(var_6.device, "/device:CPU:0")114 self.assertDeviceEqual(var_0.device, None)115 self.assertDeviceEqual(var_1.device, "/device:CPU:0")116 self.assertDeviceEqual(var_2.device, "/device:GPU:0")[all …]
1163 self.assertDeviceEqual(None, op1.device)1164 self.assertDeviceEqual("/device:GPU:0", op2.device)1165 self.assertDeviceEqual(None, op3.device)1620 self.assertDeviceEqual(None, op.device)
1325 self.assertDeviceEqual(node.device, "/job:ADevice")
3121 def assertDeviceEqual(self, device1, device2, msg=None): member in TensorFlowTestCase
766 self.assertDeviceEqual(None, var.device)767 self.assertDeviceEqual(None, var.initializer.device)773 self.assertDeviceEqual("/job:ps", var.device)774 self.assertDeviceEqual("/job:ps", var.initializer.device)
454 self.assertDeviceEqual("/job:ps", assign_op.device)
572 self.assertDeviceEqual(op.device, "/cpu:0")4405 self.assertDeviceEqual(None, with_vnod_dep.device)4413 self.assertDeviceEqual("", with_vdef_dep.device)