Searched refs:AddControlDependencyTo (Results 1 – 15 of 15) 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()
133 TF_RETURN_IF_ERROR(read->AddControlDependencyTo(write)); in AddControlEdgesForLoopWrites()
60 new_control_predecessor->AddControlDependencyTo(new_hlo_pointer)); in DefaultAction()
122 TF_CHECK_OK(add->AddControlDependencyTo(exp)); in TEST_F()
429 ASSERT_IS_OK(add->AddControlDependencyTo(negate)); in TEST_F()487 TF_CHECK_OK(negate->AddControlDependencyTo(add)); in TEST_F()
128 true_op->AddControlDependencyTo(computation->root_instruction())); in TEST_F()
156 TF_RETURN_IF_ERROR(from_copy->AddControlDependencyTo(to_copy)); in DeepCopyAndAddControlEdges()424 TF_RETURN_IF_ERROR(from->AddControlDependencyTo(to)); in AddCopiesForAliasedInputOutputs()
193 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()
1553 TF_RETURN_IF_ERROR(remat->AddControlDependencyTo(successor)); in RematerializeInstructions()1556 TF_RETURN_IF_ERROR(predecessor->AddControlDependencyTo(remat)); in RematerializeInstructions()
793 ->AddControlDependencyTo(instruction.get())); in CreateFromProto()1966 Status HloInstruction::AddControlDependencyTo(HloInstruction* instruction) { in AddControlDependencyTo() function in xla::HloInstruction2001 TF_RETURN_IF_ERROR(ctrl_pred->AddControlDependencyTo(this)); in CopyAllControlDepsFrom()2005 TF_RETURN_IF_ERROR(this->AddControlDependencyTo(ctrl_succ)); in CopyAllControlDepsFrom()
1133 TF_CHECK_OK(new_instr->AddControlDependencyTo( in CloneWithReplacements()
1108 Status AddControlDependencyTo(HloInstruction* instruction);
2632 Status status = pre->AddControlDependencyTo(instruction); in ParseInstructionRhs()