Searched refs:new_tuple (Results 1 – 6 of 6) sorted by relevance
404 HloInstruction* new_tuple = infeed_data->parent()->AddInstruction( in TEST_F() local407 TF_EXPECT_OK(infeed_data->ReplaceUseWith(user, new_tuple)); in TEST_F()429 EXPECT_TRUE(new_tuple->has_sharding()); in TEST_F()431 new_tuple->sharding(), in TEST_F()432 HloSharding::Tuple(new_tuple->shape(), {HloSharding::AssignDevice(1), in TEST_F()
275 HloInstruction* new_tuple = conditional->parent()->AddInstruction( in TryRemoveUnusedConditionalOperands() local278 conditional->ReplaceOperandWithDifferentShape(branch + 1, new_tuple)); in TryRemoveUnusedConditionalOperands()
367 HloInstruction* new_tuple = in RestructureConditionalInstruction() local370 computation->set_root_instruction(new_tuple); in RestructureConditionalInstruction()383 conditional->ReplaceUseWith(new_tuple_user, new_tuple)); in RestructureConditionalInstruction()
199 HloInstruction* new_tuple = in RemoveDeadTupleIndices() local201 TF_RETURN_IF_ERROR(computation->ReplaceInstruction(while_op, new_tuple)); in RemoveDeadTupleIndices()
389 HloInstruction* new_tuple = computation->AddInstruction( in CanonicalizeBackwardInputConvolution() local394 << new_tuple->ToString(); in CanonicalizeBackwardInputConvolution()396 TF_CHECK_OK(computation->ReplaceInstruction(backward_conv, new_tuple)); in CanonicalizeBackwardInputConvolution()
808 HloInstruction* new_tuple = in RunOnInstruction() local815 TF_RETURN_IF_ERROR(instr->parent()->ReplaceInstruction(instr, new_tuple)); in RunOnInstruction()