Searched refs:kCPU0 (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/compiler/jit/ |
D | device_util_test.cc | 59 const char* kCPU0 = "/job:localhost/replica:0/task:0/device:CPU:0"; variable 77 CheckPickDeviceResult(kCPU0, false, {kCPU0, kCPU0Partial}); in TEST() 84 CheckPickDeviceResult(kGPU0, false, {kGPU0, kCPU0}); in TEST() 85 CheckPickDeviceResult(kGPU0, false, {kCPU0, kGPU0}); in TEST() 86 CheckPickDeviceResult(kXPU0, true, {kXPU0, kCPU0}); in TEST() 102 CheckPickDeviceHasError(true, {kCPU0, kCPU1}); in TEST() 103 CheckPickDeviceHasError(false, {kCPU0, kCPU1}); in TEST() 106 CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); in TEST()
|
D | xla_cluster_util_test.cc | 98 const char* kCPU0 = "/job:localhost/replica:0/task:0/device:CPU:0"; variable 115 Scope root = Scope::NewRootScope().WithAssignedDevice(kCPU0).ExitOnError(); in TEST()
|
D | mark_for_compilation_pass_test.cc | 1361 const char* kCPU0 = "/job:worker/replica:0/task:0/device:CPU:0"; variable 1379 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kCPU0); in TEST() 1425 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kCPU0); in TEST() 1449 FindNodeByName(graph.get(), "test/b")->set_assigned_device_name(kCPU0); in TEST() 1477 ->set_assigned_device_name(kCPU0); in TEST() 1478 FindNodeByName(graph.get(), var_handle_name)->set_assigned_device_name(kCPU0); in TEST() 1562 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kCPU0); in TEST()
|
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 32 constexpr char kCPU0[] = "/job:localhost/replica:0/task:0/cpu:0"; variable 45 NodeSetUp("Node1", kConv2D, kCPU0, 6000, &node1_); in ReadyNodeManagerTest() 46 NodeSetUp("Node2", kConv2D, kCPU0, 5000, &node2_); in ReadyNodeManagerTest() 47 NodeSetUp("Node3", kConv2D, kCPU0, 4000, &node3_); in ReadyNodeManagerTest() 48 NodeSetUp("Node4", kConv2D, kCPU0, 3000, &node4_); in ReadyNodeManagerTest() 49 NodeSetUp("Node5", kConv2D, kCPU0, 2000, &node5_); in ReadyNodeManagerTest() 50 NodeSetUp("Node6", kConv2D, kCPU0, 1000, &node6_); in ReadyNodeManagerTest() 293 NodeSetUp("Node7", kConv2D, kCPU0, 5, &node7); in TEST_F() 294 NodeSetUp("Node8", kConv2D, kCPU0, 4, &node8); in TEST_F() 295 NodeSetUp("Node9", kConv2D, kCPU0, 3, &node9); in TEST_F() [all …]
|