Searched refs:ReplaceUseWith (Results 1 – 13 of 13) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_domain_isolator.cc | 50 TF_RETURN_IF_ERROR(operand->ReplaceUseWith(instruction, domain)); in RunInternal()
|
| D | copy_insertion.cc | 281 TF_RETURN_IF_ERROR(while_init->ReplaceUseWith(xla_while, while_init_copy)); in AddCopiesForWhile()
306 TF_RETURN_IF_ERROR(param->ReplaceUseWith(user, param_copy)); in AddCopiesForWhile()
329 TF_RETURN_IF_ERROR(root->ReplaceUseWith(user, deep_copy)); in AddCopiesForConditional()
380 TF_RETURN_IF_ERROR(param->ReplaceUseWith(user, copied)); in AddCopiesForAliasedInputOutputs()
1082 TF_RETURN_IF_ERROR(instruction->ReplaceUseWith(user, deep_copy)); in AddSpecialCaseCopies()
|
| D | ar_crs_combiner.cc | 329 TF_CHECK_OK(all_reduce->ReplaceUseWith(next, prev)); in RewriteGraph()
360 TF_CHECK_OK(other_operand->ReplaceUseWith(next, division)); in RewriteGraph()
|
| D | bfloat16_normalization.cc | 91 TF_RETURN_IF_ERROR(hlo->ReplaceUseWith(user, convert)); in InsertConvertAfterOutput()
259 TF_RETURN_IF_ERROR(hlo->ReplaceUseWith(user, tuple)); in HandleMultipleOutputs()
|
| D | hlo_domain_test.cc | 407 TF_EXPECT_OK(infeed_data->ReplaceUseWith(user, new_tuple)); in TEST_F()
656 TF_EXPECT_OK(tuple0->ReplaceUseWith(user, tuple1)); in TEST_F()
|
| D | hlo_reachability_test.cc | 175 ASSERT_IS_OK(constant2->ReplaceUseWith(negate, constant1)); in TEST_F()
|
| D | hlo_sharding_metadata.cc | 130 pass_through.operand->ReplaceUseWith(pass_through.user, gte)); in FixupPassThroughDomainLinks()
|
| D | hlo_instruction_test.cc | 427 ASSERT_IS_OK(foo->ReplaceUseWith(add_foofoo, bar)); in TEST_F()
462 ASSERT_IS_OK(foo->ReplaceUseWith(tuple, bar)); in TEST_F()
491 ASSERT_IS_OK(foo->ReplaceUseWith(exp, bar)); in TEST_F()
|
| D | bfloat16_propagation.cc | 750 TF_RETURN_IF_ERROR(operand->ReplaceUseWith(inst, copy)); in Run()
|
| D | hlo_rematerialization.cc | 1070 TF_RETURN_IF_ERROR(best->ReplaceUseWith(user, remat)); in RematerializeComputation()
|
| D | hlo_instruction.h | 964 Status ReplaceUseWith(HloInstruction* user, HloInstruction* new_producer);
|
| D | hlo_instructions.cc | 1259 instruction_to_merge->ReplaceUseWith(this, unfused_instructions.front())); in MergeFusionInstruction()
|
| D | hlo_instruction.cc | 1889 Status HloInstruction::ReplaceUseWith(HloInstruction* user, in ReplaceUseWith() function in xla::HloInstruction
|