Searched refs:InterconnectMap (Results 1 – 3 of 3) sorted by relevance
264 struct InterconnectMap { struct282 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps); argument293 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects,
975 const int BaseGPUDeviceFactory::InterconnectMap::kSameDeviceStrength = 1000;976 const int BaseGPUDeviceFactory::InterconnectMap::kStreamExecutorStrength = 1;1076 std::vector<InterconnectMap> interconnect_maps; in CreateDevices()1081 for (const InterconnectMap& im : interconnect_maps) { in CreateDevices()1257 se::Platform* gpu_manager, std::vector<InterconnectMap>* maps) { in GetInterconnectMaps()1261 InterconnectMap& imap = maps->at(0); in GetInterconnectMaps()1263 imap.strength = InterconnectMap::kStreamExecutorStrength; in GetInterconnectMaps()1276 int num_tf_gpus, const std::vector<InterconnectMap>& interconnects, in GetDeviceLocalities()1313 for (const InterconnectMap& imap : interconnects) { in GetDeviceLocalities()1339 ilink->set_strength(InterconnectMap::kSameDeviceStrength); in GetDeviceLocalities()
213 EXPECT_EQ(BaseGPUDeviceFactory::InterconnectMap::kSameDeviceStrength, in TEST_F()218 EXPECT_EQ(BaseGPUDeviceFactory::InterconnectMap::kSameDeviceStrength, in TEST_F()