Searched refs:ReplaceOperandWith (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/tests/ |
| D | multioutput_fusion_test.cc | 103 TF_CHECK_OK(dot->ReplaceOperandWith(0, gte0)); in RunTest2D()
104 TF_CHECK_OK(dot->ReplaceOperandWith(1, gte1)); in RunTest2D()
167 TF_CHECK_OK(sub->ReplaceOperandWith(0, gte0)); in RunTest1D()
168 TF_CHECK_OK(reshape->ReplaceOperandWith(0, gte1)); in RunTest1D()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_module_dce.cc | 80 while_body_root->ReplaceOperandWith(i, pass_thru_gte)); in RunWhileDCE()
|
| D | while_loop_constant_sinking.cc | 41 TF_RETURN_IF_ERROR(user->ReplaceOperandWith(i, new_instr)); in ReplaceUsesWhileKeepingLoopInvariance()
|
| D | dynamic_padder.cc | 175 TF_RETURN_IF_ERROR(inst->ReplaceOperandWith(operand_num, padded)); in Run()
|
| D | bfloat16_conversion_folding.cc | 85 hlo->ReplaceOperandWith(operand_index, operand->mutable_operand(0))); in FoldOperandConversion()
|
| D | reduce_precision_insertion.cc | 154 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWith(i, reduced)); in insert_on_inputs()
|
| D | bfloat16_normalization.cc | 115 TF_RETURN_IF_ERROR(hlo->ReplaceOperandWith(operand_idx, convert)); in InsertConvertBeforeOperand()
|
| D | layout_assignment.cc | 852 return instruction->ReplaceOperandWith(operand_no, operand_copy); in CopyOperandIfLayoutsDiffer()
1851 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWith(0, copy)); in ConstrainChannelLayouts()
1877 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWith(0, copy)); in ConstrainChannelLayouts()
2136 TF_RETURN_IF_ERROR(instruction->ReplaceOperandWith(operand_number, copy)); in AddCopyForOperand()
|
| D | hlo_verifier_test.cc | 122 TF_ASSERT_OK(negate->ReplaceOperandWith(0, emb_param)); in TEST_F()
|
| D | hlo_domain_test.cc | 640 TF_EXPECT_OK(tuple0->ReplaceOperandWith(1, copy0)); in TEST_F()
|
| D | hlo_module.cc | 442 outlined_instruction->ReplaceOperandWith(operand_num, *operand_slot)); in OutlineExpressionFromComputation()
|
| D | hlo_instruction.h | 975 Status ReplaceOperandWith(int64 operand_num, HloInstruction* new_operand);
|
| D | hlo_instructions.cc | 1303 TF_CHECK_OK(unfused_instruction->ReplaceOperandWith(index, new_operand)); in MergeFusionInstructionIntoMultiOutput()
1468 TF_CHECK_OK(clone->ReplaceOperandWith(operand_num, fused_param)); in CloneAndFuseInternal()
|
| D | hlo_instruction.cc | 1917 Status HloInstruction::ReplaceOperandWith(int64 operand_num, in ReplaceOperandWith() function in xla::HloInstruction
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | cudnn_conv_rewriter.cc | 451 TF_CHECK_OK(conv->ReplaceOperandWith(/*operand_no=*/1, reverse_filter)); in MatchBackwardInput()
|