Searched refs:prioritized_devices (Results 1 – 4 of 4) sorted by relevance
137 PrioritizedDeviceVector* prioritized_devices, in UpdatePrioritizedVectors() argument139 if (prioritized_devices->size() != devices.size()) { in UpdatePrioritizedVectors()141 prioritized_devices->emplace_back( in UpdatePrioritizedVectors()144 DeviceSet::SortPrioritizedDeviceVector(prioritized_devices); in UpdatePrioritizedVectors()150 for (const std::pair<Device*, int32>& p : *prioritized_devices) { in UpdatePrioritizedVectors()161 const PrioritizedDeviceVector& DeviceSet::prioritized_devices() const { in prioritized_devices() function in tensorflow::DeviceSet
105 TEST_F(DeviceSetTest, prioritized_devices) { in TEST_F() argument108 EXPECT_EQ(device_set().prioritized_devices(), in TEST_F()115 device_set().prioritized_devices(), in TEST_F()
71 const PrioritizedDeviceVector& prioritized_devices() const
289 pflr_device_set->prioritized_devices(); in SelectDevice()