Searched refs:profile_index_map (Results 1 – 4 of 4) sorted by relevance
55 HloProfileIndexMap profile_index_map(*hlo_module); in TEST_F() local57 CreateHloProfilePrinterData(profile_index_map, cost_analysis, in TEST_F()60 &profile_index_map); in TEST_F()
774 std::unique_ptr<HloProfileIndexMap> profile_index_map; in RunBackend() local786 profile_index_map = absl::make_unique<HloProfileIndexMap>(*module); in RunBackend()788 *profile_index_map, cost_analysis, entry_computation->name()); in RunBackend()795 std::move(profile_printer), std::move(profile_index_map)); in RunBackend()
203 const std::unordered_map<const T*, int64>& profile_index_map);
2700 const std::unordered_map<const T*, int64>& profile_index_map) { in GetProfileCounterCommon() argument2701 auto it = profile_index_map.find(&hlo); in GetProfileCounterCommon()2702 if (it == profile_index_map.end()) { in GetProfileCounterCommon()