Searched refs:assertDeviceEqual (Results 1 – 15 of 15) 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 …]
474 self.assertDeviceEqual("/job:dev_v0", ema.average(v0).device)475 self.assertDeviceEqual("/job:dev_v1", ema.average(v1).device)478 self.assertDeviceEqual("/job:default", ema.average(tensor2).device)
51 self.assertDeviceEqual("/job:ps/task:1", u.device)52 self.assertDeviceEqual("/job:ps/task:1", u.initializer.device)53 self.assertDeviceEqual("/job:ps/task:0", v.device)54 self.assertDeviceEqual("/job:ps/task:0", v.initializer.device)55 self.assertDeviceEqual("/job:ps/task:1", w.device)56 self.assertDeviceEqual("/job:ps/task:1", w.initializer.device)57 self.assertDeviceEqual("/job:ps/task:1", x.device)58 self.assertDeviceEqual("/job:ps/task:1", x.initializer.device)59 self.assertDeviceEqual("/job:worker", a.device)85 self.assertDeviceEqual("/job:ps/task:0", u.device)[all …]
98 self.assertDeviceEqual(var_0.device, None)99 self.assertDeviceEqual(var_1.device, "/device:CPU:0")100 self.assertDeviceEqual(var_2.device, None)101 self.assertDeviceEqual(var_3.device, None)102 self.assertDeviceEqual(var_4.device, "/device:CPU:0")103 self.assertDeviceEqual(var_5.device, "/device:GPU:0")104 self.assertDeviceEqual(var_6.device, "/device:CPU:0")117 self.assertDeviceEqual(var_0.device, None)118 self.assertDeviceEqual(var_1.device, "/device:CPU:0")119 self.assertDeviceEqual(var_2.device, "/device:GPU:0")[all …]
685 self.assertDeviceEqual(None, op1.device)686 self.assertDeviceEqual("/device:GPU:0", op2.device)687 self.assertDeviceEqual(None, op3.device)1130 self.assertDeviceEqual(None, op.device)
1332 self.assertDeviceEqual(node.device, "/job:ADevice")
2511 def assertDeviceEqual(self, device1, device2, msg=None): member in TensorFlowTestCase
394 self.assertDeviceEqual(loss.device, a.device)402 self.assertDeviceEqual(loss.device, a.device)409 self.assertDeviceEqual(a.device, 'cpu:0')410 self.assertDeviceEqual(b.device, 'cpu:1')417 self.assertDeviceEqual(a.device, 'cpu:0')418 self.assertDeviceEqual(b.device, 'cpu:1')440 self.assertDeviceEqual(a.device, 'cpu:0')443 self.assertDeviceEqual(b.device, 'cpu:1')446 self.assertDeviceEqual(c.device, 'cpu:12')449 self.assertDeviceEqual(d.device, 'cpu:2')[all …]
192 self.assertDeviceEqual("/job:worker/task:0", v.device)193 self.assertDeviceEqual("job:ps/task:0", v_g.device)194 self.assertDeviceEqual("/job:worker/task:0", w.device)195 self.assertDeviceEqual("job:ps/task:1", w_g.device)
274 self.assertDeviceEqual("/job:worker/task:0", v.device)275 self.assertDeviceEqual("job:ps/task:0", v_g.device)276 self.assertDeviceEqual("/job:worker/task:0", w.device)277 self.assertDeviceEqual("job:ps/task:1", w_g.device)
290 self.assertDeviceEqual("/job:worker/task:0", v.device)291 self.assertDeviceEqual("job:ps/task:0", v_g.device)292 self.assertDeviceEqual("/job:worker/task:0", w.device)293 self.assertDeviceEqual("job:ps/task:1", w_g.device)
1341 self.assertDeviceEqual('/job:ps/task:0', v.device)1342 self.assertDeviceEqual('/job:ps/task:0', v.initializer.device)1343 self.assertDeviceEqual('/job:ps/task:0', w.device)1344 self.assertDeviceEqual('/job:ps/task:0', w.initializer.device)1345 self.assertDeviceEqual('/job:worker', a.device)1353 self.assertDeviceEqual('', v.device)1354 self.assertDeviceEqual('', v.initializer.device)1355 self.assertDeviceEqual('', w.device)1356 self.assertDeviceEqual('', w.initializer.device)1357 self.assertDeviceEqual('', a.device)[all …]
617 self.assertDeviceEqual(None, var.device)618 self.assertDeviceEqual(None, var.initializer.device)624 self.assertDeviceEqual("/job:ps", var.device)625 self.assertDeviceEqual("/job:ps", var.initializer.device)
426 self.assertDeviceEqual("/job:ps", assign_op.device)
541 self.assertDeviceEqual(op.device, "/cpu:0")3969 self.assertDeviceEqual(None, with_vnod_dep.device)3977 self.assertDeviceEqual("", with_vdef_dep.device)