Home
last modified time | relevance | path

Searched refs:kCPU1 (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util_test.cc64 const char* kCPU1 = "/job:localhost/replica:0/task:0/device:CPU:1"; variable
98 CheckPickDeviceHasError(false, {kXPU0, kCPU1}); in TEST()
102 CheckPickDeviceHasError(true, {kCPU0, kCPU1}); in TEST()
103 CheckPickDeviceHasError(false, {kCPU0, kCPU1}); in TEST()
106 CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); in TEST()
/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc33 constexpr char kCPU1[] = "/job:localhost/replica:0/task:0/cpu:1"; variable
475 NodeSetUp("Node7", kConv2D, kCPU1, 1001, &node7); in TEST_F()
476 NodeSetUp("Node8", kConv2D, kCPU1, 2001, &node8); in TEST_F()
477 NodeSetUp("Node9", kConv2D, kCPU1, 3001, &node9); in TEST_F()
487 NodeSetUp("Recv2", kRecv, kCPU1, 2004, &recv2); in TEST_F()
565 NodeSetUp("Node12", kConv2D, kCPU1, 1000, &node12); in TEST_F()
676 devices[kCPU1] = cpu_device; in VirtualSchedulerTest()
2230 auto y1 = ops::Identity(s.WithOpName("y1").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
2231 auto y2 = ops::Identity(s.WithOpName("y2").WithDevice(kCPU1), y); in CreateGrapplerItemWithInterDeviceTransfers()
2235 s.WithOpName("batch_mean1").WithDevice(kCPU1), batch_mean); in CreateGrapplerItemWithInterDeviceTransfers()
[all …]