Home
last modified time | relevance | path

Searched refs:assertDeviceEqual (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Ddevice_setter_test.py46 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 …]
Dmoving_averages_test.py474 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)
/external/tensorflow/tensorflow/contrib/training/python/training/
Ddevice_setter_test.py51 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 …]
/external/tensorflow/tensorflow/python/framework/
Dgraph_util_test.py98 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 …]
Dops_test.py685 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)
Dop_def_library_test.py1332 self.assertDeviceEqual(node.device, "/job:ADevice")
Dtest_util.py2511 def assertDeviceEqual(self, device1, device2, msg=None): member in TensorFlowTestCase
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py394 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 …]
/external/tensorflow/tensorflow/contrib/opt/python/training/
Dmodel_average_optimizer_test.py192 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)
Dagn_optimizer_test.py274 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)
Delastic_average_optimizer_test.py290 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)
/external/tensorflow/tensorflow/contrib/learn/python/learn/estimators/
Destimator_test.py1341 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 …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dinit_ops_test.py617 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)
Dvariables_test.py426 self.assertDeviceEqual("/job:ps", assign_op.device)
Dcontrol_flow_ops_py_test.py541 self.assertDeviceEqual(op.device, "/cpu:0")
3969 self.assertDeviceEqual(None, with_vnod_dep.device)
3977 self.assertDeviceEqual("", with_vdef_dep.device)