Searched refs:kCPU1 (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | device_util_test.cc | 64 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/ |
D | virtual_scheduler_test.cc | 33 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 …]
|