Home
last modified time | relevance | path

Searched refs:AddControlDependencyTo (Results 1 – 14 of 14) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dcall_inliner_test.cc53 TF_ASSERT_OK(zero->AddControlDependencyTo(one)); in TEST_F()
130 TF_ASSERT_OK(false_constant->AddControlDependencyTo(true_constant)); in TEST_F()
Dcall_inliner.cc61 new_control_predecessor->AddControlDependencyTo(new_hlo_pointer)); in DefaultAction()
Dconditional_simplifier_test.cc113 true_op->AddControlDependencyTo(computation->root_instruction())); in TEST_F()
Dhlo_reachability_test.cc122 TF_CHECK_OK(add->AddControlDependencyTo(exp)); in TEST_F()
Dhlo_computation_test.cc430 ASSERT_IS_OK(add->AddControlDependencyTo(negate)); in TEST_F()
487 TF_CHECK_OK(negate->AddControlDependencyTo(add)); in TEST_F()
Dhlo_rematerialization.cc1056 TF_RETURN_IF_ERROR(remat->AddControlDependencyTo(successor)); in RematerializeComputation()
1059 TF_RETURN_IF_ERROR(predecessor->AddControlDependencyTo(remat)); in RematerializeComputation()
Dcopy_insertion.cc156 TF_RETURN_IF_ERROR(from_copy->AddControlDependencyTo(to_copy)); in DeepCopyAndAddControlEdges()
412 TF_RETURN_IF_ERROR(from->AddControlDependencyTo(to)); in AddCopiesForAliasedInputOutputs()
Dhlo_dce_test.cc155 TF_ASSERT_OK(dead_negate_with_control_dep->AddControlDependencyTo( in TEST_F()
Dwhile_loop_simplifier_test.cc172 TF_ASSERT_OK(true_op->AddControlDependencyTo( in TEST_F()
Dwhile_loop_invariant_code_motion_test.cc423 TF_ASSERT_OK(param->AddControlDependencyTo(add_result)); in TEST_F()
Dhlo_instruction.cc614 ->AddControlDependencyTo(instruction.get())); in CreateFromProto()
1630 Status HloInstruction::AddControlDependencyTo(HloInstruction* instruction) { in AddControlDependencyTo() function in xla::HloInstruction
1665 TF_RETURN_IF_ERROR(ctrl_pred->AddControlDependencyTo(this)); in CopyAllControlDepsFrom()
1669 TF_RETURN_IF_ERROR(this->AddControlDependencyTo(ctrl_succ)); in CopyAllControlDepsFrom()
Dhlo_computation.cc1031 TF_CHECK_OK(new_instr->AddControlDependencyTo( in CloneWithReplacements()
Dhlo_instruction.h868 Status AddControlDependencyTo(HloInstruction* instruction);
Dhlo_parser.cc1739 Status status = pre->AddControlDependencyTo(instruction); in ParseInstructionRhs()