Home
last modified time | relevance | path

Searched refs:kCPU0 (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/jit/
Ddevice_util_test.cc59 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()
Dxla_cluster_util_test.cc98 const char* kCPU0 = "/job:localhost/replica:0/task:0/device:CPU:0"; variable
115 Scope root = Scope::NewRootScope().WithAssignedDevice(kCPU0).ExitOnError(); in TEST()
Dmark_for_compilation_pass_test.cc1361 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/
Dvirtual_scheduler_test.cc32 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 …]