Searched refs:ReplaceUseWith (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | copy_insertion.cc | 304 TF_RETURN_IF_ERROR(while_init->ReplaceUseWith(xla_while, while_init_copy)); in AddCopiesForWhile()
328 TF_RETURN_IF_ERROR(param->ReplaceUseWith(user, param_copy)); in AddCopiesForWhile()
344 TF_RETURN_IF_ERROR(operand->ReplaceUseWith(in_place_op, deep_copy)); in AddCopiesForInPlaceOperation()
392 TF_RETURN_IF_ERROR(param->ReplaceUseWith(user, copied)); in AddCopiesForAliasedInputOutputs()
1003 TF_RETURN_IF_ERROR(root->ReplaceUseWith(user, deep_copy)); in AddCopiesForConditional()
1156 TF_RETURN_IF_ERROR(instruction->ReplaceUseWith(user, deep_copy)); in AddSpecialCaseCopies()
|
| 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 | ar_crs_combiner.cc | 548 TF_CHECK_OK(all_reduce->ReplaceUseWith(next, prev)); in RewriteGraph()
579 TF_CHECK_OK(other_operand->ReplaceUseWith(next, division)); in RewriteGraph()
|
| D | hlo_reachability_test.cc | 175 ASSERT_IS_OK(constant2->ReplaceUseWith(negate, constant1)); in TEST_F()
|
| D | dynamic_padder.cc | 484 TF_RETURN_IF_ERROR(reshape->ReplaceUseWith(user, reshape_dynamic)); in RewriteDynamicReshapeSplitInput()
690 TF_RETURN_IF_ERROR(reshape->ReplaceUseWith(user, gather)); in RewriteDynamicReshapeCombineInput()
1273 TF_RETURN_IF_ERROR(sort->ReplaceUseWith(user, rewritten_sort)); in RewriteDynamicSort()
1534 TF_RETURN_IF_ERROR(param->ReplaceUseWith(user, static_param)); in InsertPadToStaticAfterModuleInputs()
|
| D | hlo_sharding_metadata.cc | 130 pass_through.operand->ReplaceUseWith(pass_through.user, gte)); in FixupPassThroughDomainLinks()
|
| D | bfloat16_normalization.cc | 373 TF_RETURN_IF_ERROR(hlo->ReplaceUseWith(user, tuple)); in HandleMultipleOutputs()
|
| D | hlo_instruction_test.cc | 429 ASSERT_IS_OK(foo->ReplaceUseWith(add_foofoo, bar)); in TEST_F()
464 ASSERT_IS_OK(foo->ReplaceUseWith(tuple, bar)); in TEST_F()
493 ASSERT_IS_OK(foo->ReplaceUseWith(exp, bar)); in TEST_F()
|
| D | bfloat16_propagation.cc | 870 TF_RETURN_IF_ERROR(operand->ReplaceUseWith(inst, copy)); in Run()
|
| D | hlo_instruction.h | 1186 Status ReplaceUseWith(HloInstruction* user, HloInstruction* new_producer);
|
| D | conditional_code_motion.cc | 383 conditional->ReplaceUseWith(new_tuple_user, new_tuple)); in RestructureConditionalInstruction()
|
| D | hlo_rematerialization.cc | 1711 TF_RETURN_IF_ERROR(best->ReplaceUseWith(user, uncompressed)); in CompressInstruction()
|
| D | hlo_instructions.cc | 1500 TF_CHECK_OK(instruction_to_merge->ReplaceUseWith(this, unfused_root)); in MergeFusionInstruction()
|
| D | hlo_instruction.cc | 2300 Status HloInstruction::ReplaceUseWith(HloInstruction* user, in ReplaceUseWith() function in xla::HloInstruction
|