Searched refs:kCPU0 (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/costs/ |
D | virtual_scheduler_test.cc | 55 NodeSetUp("Node1", kConv2D, kCPU0, 6000, &node1_); in VirtualSchedulerTest() 56 NodeSetUp("Node2", kConv2D, kCPU0, 5000, &node2_); in VirtualSchedulerTest() 57 NodeSetUp("Node3", kConv2D, kCPU0, 4000, &node3_); in VirtualSchedulerTest() 58 NodeSetUp("Node4", kConv2D, kCPU0, 3000, &node4_); in VirtualSchedulerTest() 59 NodeSetUp("Node5", kConv2D, kCPU0, 2000, &node5_); in VirtualSchedulerTest() 60 NodeSetUp("Node6", kConv2D, kCPU0, 1000, &node6_); in VirtualSchedulerTest() 70 devices[kCPU0] = cpu_device; in VirtualSchedulerTest() 82 const string kCPU0 = "/job:localhost/replica:0/task:0/cpu:0"; member in tensorflow::grappler::VirtualSchedulerTest 117 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithConv2Ds() 144 Scope s = Scope::NewRootScope().WithDevice(kCPU0); in CreateGrapplerItemWithConv2DAndVariable() [all …]
|
/external/tensorflow/tensorflow/compiler/jit/ |
D | xla_cluster_util_test.cc | 120 const char* kCPU0 = "/job:localhost/replica:0/task:0/device:CPU:0"; variable 133 CheckPickDeviceResult(kGPU0, false, {kGPU0, kCPU0}); in TEST() 134 CheckPickDeviceResult(kXPU0, true, {kXPU0, kCPU0}); in TEST() 150 CheckPickDeviceHasError(false, {kCPU0, kCPU1}); in TEST() 153 CheckPickDeviceHasError(false, {kCPU0, kCPU1, kGPU0}); in TEST()
|
D | mark_for_compilation_pass_test.cc | 1190 const char* kCPU0 = "/job:worker/replica:0/task:0/device:CPU:0"; variable 1208 FindNodeByName(graph.get(), "test/y")->set_assigned_device_name(kCPU0); in TEST() 1254 FindNodeByName(graph.get(), "test/x")->set_assigned_device_name(kCPU0); in TEST() 1278 FindNodeByName(graph.get(), "test/b")->set_assigned_device_name(kCPU0); in TEST() 1306 ->set_assigned_device_name(kCPU0); in TEST() 1307 FindNodeByName(graph.get(), var_handle_name)->set_assigned_device_name(kCPU0); in TEST()
|