Lines Matching refs:node
38 NodeDef node;
39 node.set_op("Conv2D");
40 // node.device() is empty, but GPU is default device if there is.
41 EXPECT_EQ("GPU", placer.get_device(node).type());
43 placer.get_canonical_device_name(node));
45 node.set_device("CPU");
46 EXPECT_EQ("CPU", placer.get_device(node).type());
48 placer.get_canonical_device_name(node));
50 node.set_device("GPU:0");
51 EXPECT_EQ("GPU", placer.get_device(node).type());
53 placer.get_canonical_device_name(node));
68 NodeDef node;
69 node.set_op("Conv2D");
70 // node.device() is empty, but GPU is default device if there is.
71 EXPECT_EQ("GPU", placer.get_device(node).type());
72 EXPECT_EQ("/GPU:0", placer.get_canonical_device_name(node));
74 node.set_device("CPU");
75 EXPECT_EQ("CPU", placer.get_device(node).type());
76 EXPECT_EQ("/CPU:0", placer.get_canonical_device_name(node));
78 node.set_device("GPU:0");
79 EXPECT_EQ("GPU", placer.get_device(node).type());
80 EXPECT_EQ("/GPU:0", placer.get_canonical_device_name(node));
98 NodeDef node;
99 node.set_op("Conv2D");
100 // node.device() is empty, and CPU is default device.
101 EXPECT_EQ("CPU", placer.get_device(node).type());
103 placer.get_canonical_device_name(node));
105 node.set_device("/device:TPU:0");
106 EXPECT_EQ("TPU", placer.get_device(node).type());
108 placer.get_canonical_device_name(node));
128 NodeDef node;
129 node.set_op("Conv2D");
130 node.set_device("/device:CPU:0");
132 placer.get_canonical_device_name(node));
133 node.set_device("/device:GPU:0");
136 placer.get_canonical_device_name(node));
150 NodeDef node;
151 node.set_op("Conv2D");
152 node.set_device("/device:CPU:0");
154 placer.get_canonical_device_name(node));
161 NodeDef node;
162 node.set_op("Conv2D");
163 // Device is not set to the node, so get_canonical_device_name() will return
165 return placer.get_canonical_device_name(node);
210 NodeDef node;
211 node.set_op("Conv2D");
212 node.set_device(device);
213 return placer->get_canonical_device_name(node);
260 NodeDef node;
261 node.set_op("Conv2D");
264 EXPECT_EQ("UNKNOWN", placer.get_device(node).type());
265 EXPECT_EQ("UNKNOWN", placer.get_canonical_device_name(node));
276 NodeDef node;
277 node.set_op("Conv2D");
280 EXPECT_EQ("CPU", placer.get_device(node).type());
282 placer.get_canonical_device_name(node));
296 NodeDef node;
297 node.set_op("Conv2D");
300 EXPECT_EQ("GPU", placer.get_device(node).type());
302 placer.get_canonical_device_name(node));
304 node.set_device("/job:my_job/replica:0/task:0/cpu:0");
305 EXPECT_EQ("CPU", placer.get_device(node).type());
307 placer.get_canonical_device_name(node));
309 node.set_device("/job:my_job/replica:0/task:0/device:GPU:0");
310 EXPECT_EQ("GPU", placer.get_device(node).type());
312 placer.get_canonical_device_name(node));
315 node.set_device("CPU");
316 EXPECT_EQ("GPU", placer.get_device(node).type());
318 placer.get_canonical_device_name(node));
320 node.set_device("GPU:0");
322 EXPECT_EQ("GPU", placer.get_device(node).type());
324 placer.get_canonical_device_name(node));
327 node.set_device("/job:my_job/replica:0/task:0");
328 EXPECT_EQ("GPU", placer.get_device(node).type());
330 placer.get_canonical_device_name(node));
