Searched refs:PrioritizedDeviceVector (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_set.h | 30 typedef std::vector<std::pair<Device*, int32>> PrioritizedDeviceVector; typedef
71 const PrioritizedDeviceVector& prioritized_devices() const
95 static void SortPrioritizedDeviceVector(PrioritizedDeviceVector* vector);
114 mutable PrioritizedDeviceVector prioritized_devices_
|
| D | device_set_test.cc | 109 (PrioritizedDeviceVector{std::make_pair(d2, 51), in TEST_F()
116 (PrioritizedDeviceVector{std::make_pair(d2, 51), std::make_pair(d1, 50), in TEST_F()
145 PrioritizedDeviceVector sorted{ in TEST_F()
152 EXPECT_EQ(sorted, (PrioritizedDeviceVector{ in TEST_F()
|
| D | device_set.cc | 107 void DeviceSet::SortPrioritizedDeviceVector(PrioritizedDeviceVector* vector) { in SortPrioritizedDeviceVector()
137 PrioritizedDeviceVector* prioritized_devices, in UpdatePrioritizedVectors()
161 const PrioritizedDeviceVector& DeviceSet::prioritized_devices() const { in prioritized_devices()
|
| D | colocation_graph.cc | 1483 PrioritizedDeviceVector prioritized_filtered_devices; in FilterSupportedDevices()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context.cc | 221 std::vector<string> DevicesToString(const PrioritizedDeviceVector& devices) { in DevicesToString()
250 const PrioritizedDeviceVector& existing, in SelectBestMatchingDevice()
288 const PrioritizedDeviceVector& existing = in SelectDevice()
|