Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dvirtual_scheduler_test.cc55 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/
Dxla_cluster_util_test.cc120 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()
Dmark_for_compilation_pass_test.cc1190 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()