Searched refs:TfToPlatformGpuIdMap (Results 1 – 1 of 1) sorted by relevance
30 class TfToPlatformGpuIdMap { class32 static TfToPlatformGpuIdMap* singleton() { in singleton()33 static auto* id_map = new TfToPlatformGpuIdMap; in singleton()70 TfToPlatformGpuIdMap() = default;82 TF_DISALLOW_COPY_AND_ASSIGN(TfToPlatformGpuIdMap);88 return TfToPlatformGpuIdMap::singleton()->Insert(tf_gpu_id, platform_gpu_id); in InsertTfPlatformGpuIdPair()93 if (TfToPlatformGpuIdMap::singleton()->Find(tf_gpu_id, platform_gpu_id)) { in TfToPlatformGpuId()101 TfToPlatformGpuIdMap::singleton()->TestOnlyReset(); in TestOnlyReset()