Lines Matching refs:devices
58 std::vector<tensorflow::Device*> devices;
60 opts, kDeviceNamePrefix, &devices);
68 std::vector<tensorflow::Device*> devices;
70 opts, kDeviceNamePrefix, &devices);
79 std::vector<tensorflow::Device*> devices;
81 opts, kDeviceNamePrefix, &devices);
90 std::vector<tensorflow::Device*> devices;
92 opts, kDeviceNamePrefix, &devices);
103 std::vector<tensorflow::Device*> devices;
105 opts, kDeviceNamePrefix, &devices);
108 "Not enough GPUs to create virtual devices."))
116 std::vector<tensorflow::Device*> devices;
118 opts, kDeviceNamePrefix, &devices);
121 "Not enough GPUs to create virtual devices."))
131 std::vector<tensorflow::Device*> devices;
133 opts, kDeviceNamePrefix, &devices);
145 std::vector<tensorflow::Device*> devices;
147 opts, kDeviceNamePrefix, &devices));
148 EXPECT_EQ(1, devices.size());
149 EXPECT_GE(devices[0]->attributes().memory_limit(), 0);
150 for (auto d : devices) delete d;
157 std::vector<tensorflow::Device*> devices;
159 opts, kDeviceNamePrefix, &devices));
160 EXPECT_EQ(1, devices.size());
161 EXPECT_GE(devices[0]->attributes().memory_limit(), 0);
162 for (auto d : devices) delete d;
167 std::vector<tensorflow::Device*> devices;
169 opts, kDeviceNamePrefix, &devices));
170 EXPECT_EQ(1, devices.size());
171 EXPECT_EQ(123 << 20, devices[0]->attributes().memory_limit());
172 for (auto d : devices) delete d;
177 std::vector<tensorflow::Device*> devices;
179 opts, kDeviceNamePrefix, &devices));
180 EXPECT_EQ(2, devices.size());
181 EXPECT_EQ(123 << 20, devices[0]->attributes().memory_limit());
182 EXPECT_EQ(456 << 20, devices[1]->attributes().memory_limit());
183 ASSERT_EQ(1, devices[0]->attributes().locality().links().link_size());
184 ASSERT_EQ(1, devices[1]->attributes().locality().links().link_size());
185 EXPECT_EQ(1, devices[0]->attributes().locality().links().link(0).device_id());
187 devices[0]->attributes().locality().links().link(0).type());
189 devices[0]->attributes().locality().links().link(0).strength());
190 EXPECT_EQ(0, devices[1]->attributes().locality().links().link(0).device_id());
192 devices[1]->attributes().locality().links().link(0).type());
194 devices[1]->attributes().locality().links().link(0).strength());
195 for (auto d : devices) delete d;
