Searched refs:add_new_instr (Results 1 – 1 of 1) sorted by relevance
357 auto add_new_instr = [&](std::unique_ptr<HloInstruction> instr) { in TryRemoveConstantParams() local370 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams()387 add_new_instr(HloInstruction::CreateGetTupleElement( in TryRemoveConstantParams()413 add_constant_elems(add_new_instr(HloInstruction::CreateParameter( in TryRemoveConstantParams()422 add_constant_elems(add_new_instr(HloInstruction::CreateParameter( in TryRemoveConstantParams()429 add_new_instr(while_body->root_instruction()->Clone())), in TryRemoveConstantParams()442 add_new_instr(remove_constant_elems(while_init))))))); in TryRemoveConstantParams()676 auto add_new_instr = [&](std::unique_ptr<HloInstruction> instr) { in TryFlattenNestedTuples() local685 gtes.push_back(add_new_instr(HloInstruction::CreateGetTupleElement( in TryFlattenNestedTuples()705 nested(add_new_instr(HloInstruction::CreateParameter( in TryFlattenNestedTuples()[all …]