Searched refs:new_tuple (Results 1 – 5 of 5) 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()
301 HloInstruction* new_tuple = in TryRemoveDeadWhileParams() local303 TF_RETURN_IF_ERROR(while_op->ReplaceAllUsesWith(new_tuple)); in TryRemoveDeadWhileParams()
145 std::unique_ptr<HloInstruction> new_tuple = HloInstruction::CreateTuple({ in HandleBatchNormTraining() local154 batch_norm, std::move(new_tuple))); in HandleBatchNormTraining()
381 HloInstruction* new_tuple = computation->AddInstruction( in CanonicalizeBackwardInputConvolution() local386 << new_tuple->ToString(); in CanonicalizeBackwardInputConvolution()388 TF_CHECK_OK(computation->ReplaceInstruction(backward_conv, new_tuple)); in CanonicalizeBackwardInputConvolution()
398 HloInstruction* new_tuple = in RunOnInstruction() local405 TF_RETURN_IF_ERROR(instr->parent()->ReplaceInstruction(instr, new_tuple)); in RunOnInstruction()