Searched refs:kCPU1 (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_cluster_util_test.cc | 124 const char* kCPU1 = "/job:localhost/replica:0/task:0/device:CPU:1"; variable
146 CheckPickDeviceHasError(false, {kXPU0, kCPU1}); in TEST()
150 CheckPickDeviceHasError(false, {kCPU0, kCPU1}); in TEST()
153 CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); in TEST()
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler_test.cc | 71 devices[kCPU1] = cpu_device; in VirtualSchedulerTest()
83 const string kCPU1 = "/job:localhost/replica:0/task:0/cpu:1"; member in tensorflow::grappler::VirtualSchedulerTest
1564 auto y1 = ops::Identity(s.WithOpName("y1").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
1565 auto y2 = ops::Identity(s.WithOpName("y2").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
1569 s.WithOpName("batch_mean1").WithDevice(kCPU1), batch_mean); in CreateGrapplerItemWithInterDeviceTransfers()
1571 ops::Identity(s.WithOpName("batch_var1").WithDevice(kCPU1), batch_var); in CreateGrapplerItemWithInterDeviceTransfers()
1575 .WithDevice(kCPU1)); in CreateGrapplerItemWithInterDeviceTransfers()
2110 NodeSetUp("Node7", kConv2D, kCPU1, 1001, &node7); in TEST_F()
2111 NodeSetUp("Node8", kConv2D, kCPU1, 2001, &node8); in TEST_F()
2112 NodeSetUp("Node9", kConv2D, kCPU1, 3001, &node9); in TEST_F()
[all …]
|