Searched refs:AddControlDependencyTo (Results 1 – 14 of 14) sorted by relevance
53 TF_ASSERT_OK(zero->AddControlDependencyTo(one)); in TEST_F()130 TF_ASSERT_OK(false_constant->AddControlDependencyTo(true_constant)); in TEST_F()
61 new_control_predecessor->AddControlDependencyTo(new_hlo_pointer)); in DefaultAction()
113 true_op->AddControlDependencyTo(computation->root_instruction())); in TEST_F()
122 TF_CHECK_OK(add->AddControlDependencyTo(exp)); in TEST_F()
430 ASSERT_IS_OK(add->AddControlDependencyTo(negate)); in TEST_F()487 TF_CHECK_OK(negate->AddControlDependencyTo(add)); in TEST_F()
1056 TF_RETURN_IF_ERROR(remat->AddControlDependencyTo(successor)); in RematerializeComputation()1059 TF_RETURN_IF_ERROR(predecessor->AddControlDependencyTo(remat)); in RematerializeComputation()
156 TF_RETURN_IF_ERROR(from_copy->AddControlDependencyTo(to_copy)); in DeepCopyAndAddControlEdges()412 TF_RETURN_IF_ERROR(from->AddControlDependencyTo(to)); in AddCopiesForAliasedInputOutputs()
155 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()
614 ->AddControlDependencyTo(instruction.get())); in CreateFromProto()1630 Status HloInstruction::AddControlDependencyTo(HloInstruction* instruction) { in AddControlDependencyTo() function in xla::HloInstruction1665 TF_RETURN_IF_ERROR(ctrl_pred->AddControlDependencyTo(this)); in CopyAllControlDepsFrom()1669 TF_RETURN_IF_ERROR(this->AddControlDependencyTo(ctrl_succ)); in CopyAllControlDepsFrom()
1031 TF_CHECK_OK(new_instr->AddControlDependencyTo( in CloneWithReplacements()
868 Status AddControlDependencyTo(HloInstruction* instruction);
1739 Status status = pre->AddControlDependencyTo(instruction); in ParseInstructionRhs()