Searched refs:TryFindHostDevice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | pin_to_host_optimizer_test.cc | 34 EXPECT_EQ(internal::TryFindHostDevice(devices, false, "ABC"), ""); in TEST_F()
40 EXPECT_EQ(internal::TryFindHostDevice(devices, true, ""), "/device:CPU:0"); in TEST_F()
41 EXPECT_EQ(internal::TryFindHostDevice(devices, true, "/device:XLA_GPU:0"), in TEST_F()
43 EXPECT_EQ(internal::TryFindHostDevice(devices, true, "/device:XLA_GPU:*"), in TEST_F()
|
| D | pin_to_host_optimizer.h | 29 string TryFindHostDevice(const gtl::FlatSet<string>& devices,
|
| D | pin_to_host_optimizer.cc | 273 string TryFindHostDevice(const gtl::FlatSet<string>& devices, in TryFindHostDevice() function
338 internal::TryFindHostDevice(devices, has_device_cpu, node.device()); in Optimize()
|