Searched refs:remat_move_instructions (Results 1 – 1 of 1) sorted by relevance
1527 absl::flat_hash_set<const HloInstruction*>* remat_move_instructions, in RematerializeInstructions() argument1660 if (ContainsKey(*remat_move_instructions, best)) { in RematerializeInstructions()1667 remat_move_instructions->insert(remat); in RematerializeInstructions()1753 absl::flat_hash_set<const HloInstruction*>* remat_move_instructions) { in RematerializeBestBlock() argument1786 remat_move_instructions, instruction_list)); in RematerializeBestBlock()1854 absl::flat_hash_set<const HloInstruction*> remat_move_instructions; in RematerializeComputation() local1907 &remat_move_instructions)); in RematerializeComputation()