Searched refs:AddControlDependencyTo (Results 1 – 18 of 18) sorted by relevance
52 TF_ASSERT_OK(zero->AddControlDependencyTo(one)); in TEST_F()129 TF_ASSERT_OK(false_constant->AddControlDependencyTo(true_constant)); in TEST_F()
59 TF_RETURN_IF_ERROR(prev->AddControlDependencyTo(next)); in Run()
166 ->AddControlDependencyTo( in TEST_F()
133 TF_RETURN_IF_ERROR(read->AddControlDependencyTo(write)); in AddControlEdgesForLoopWrites()
63 new_control_predecessor->AddControlDependencyTo(new_hlo_pointer)); in DefaultAction()
411 ASSERT_IS_OK(add->AddControlDependencyTo(negate)); in TEST_F()469 TF_CHECK_OK(negate->AddControlDependencyTo(add)); in TEST_F()
165 TF_RETURN_IF_ERROR(from_copy->AddControlDependencyTo(to_copy)); in DeepCopyAndAddControlEdges()438 TF_RETURN_IF_ERROR(from->AddControlDependencyTo(to)); in AddCopiesForAliasedInputOutputs()833 TF_CHECK_OK(entry2->AddControlDependencyTo(entry1)); in AddControlDependenceForUnorderedOps()
123 TF_CHECK_OK(add->AddControlDependencyTo(exp)); in TEST_F()
127 true_op->AddControlDependencyTo(computation->root_instruction())); in TEST_F()
191 TF_ASSERT_OK(dead_negate_with_control_dep->AddControlDependencyTo( in TEST_F()
172 TF_ASSERT_OK(true_op->AddControlDependencyTo( in TEST_F()
423 TF_ASSERT_OK(param->AddControlDependencyTo(add_result)); in TEST_F()
1582 TF_RETURN_IF_ERROR(remat->AddControlDependencyTo(successor)); in RematerializeInstructions()1585 TF_RETURN_IF_ERROR(predecessor->AddControlDependencyTo(remat)); in RematerializeInstructions()
977 ->AddControlDependencyTo(instruction.get())); in CreateFromProto()2305 Status HloInstruction::AddControlDependencyTo(HloInstruction* instruction) { in AddControlDependencyTo() function in xla::HloInstruction2340 TF_RETURN_IF_ERROR(ctrl_pred->AddControlDependencyTo(this)); in CopyAllControlDepsFrom()2344 TF_RETURN_IF_ERROR(this->AddControlDependencyTo(ctrl_succ)); in CopyAllControlDepsFrom()
1308 TF_CHECK_OK(new_instr->AddControlDependencyTo( in CloneWithReplacements()
1247 Status AddControlDependencyTo(HloInstruction* instruction);
1223 Status status = pre->AddControlDependencyTo(instruction); in ParseInstructionRhs()
132 TF_CHECK_OK(add1->AddControlDependencyTo(nonblocking_call)); in TEST_F()209 TF_CHECK_OK(add1->AddControlDependencyTo(all_reduce_start)); in TEST_F()