Searched refs:remat_use (Results 1 – 1 of 1) sorted by relevance
1569 HloInstruction* remat_use = remat; in RematerializeInstructions() local1573 remat_use = computation->AddInstruction( in RematerializeInstructions()1575 ShapeUtil::GetTupleElementShape(remat_use->shape(), in RematerializeInstructions()1577 remat_use, *user.index)); in RematerializeInstructions()1578 indirect_users.push_back(instruction_list->CreateItem(remat_use)); in RematerializeInstructions()1579 gte_cache[*user.index] = remat_use; in RematerializeInstructions()1581 remat_use = cached_gte->second; in RematerializeInstructions()1585 remat_use->shape()) { in RematerializeInstructions()1586 remat_use = computation->AddInstruction(HloInstruction::CreateBitcast( in RematerializeInstructions()1587 user.user->instruction->operand(op_idx)->shape(), remat_use)); in RematerializeInstructions()[all …]