Searched refs:thunk_to_hlo (Results 1 – 3 of 3) sorted by relevance
53 absl::flat_hash_map<const Thunk*, const HloInstruction*> thunk_to_hlo) in ThunkSchedule() argument56 thunk_to_hlo_(std::move(thunk_to_hlo)) { in ThunkSchedule()
56 absl::flat_hash_map<const Thunk*, const HloInstruction*> thunk_to_hlo);
611 absl::flat_hash_map<const Thunk*, const HloInstruction*> thunk_to_hlo; in CompileModuleToLlvmIrImpl() local635 InsertOrDie(&thunk_to_hlo, thunk.get(), instruction); in CompileModuleToLlvmIrImpl()655 std::move(stream_assignment), std::move(thunk_to_hlo)); in CompileModuleToLlvmIrImpl()