Home
last modified time | relevance | path

Searched refs:SetupDerivedInstruction (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dall_to_all_decomposer.cc86 instruction->SetupDerivedInstruction(operand_reshape); in ExpandInstruction()
92 instruction->SetupDerivedInstruction(output_reshape); in ExpandInstruction()
109 all_to_all->SetupDerivedInstruction(slices.back()); in ExpandInstruction()
124 all_to_all->SetupDerivedInstruction(new_all_to_all); in ExpandInstruction()
128 all_to_all->SetupDerivedInstruction(concat); in ExpandInstruction()
H A Dconditional_to_select.cc49 conditional->SetupDerivedInstruction(if_call_op); in DoConditionalToSelect()
54 conditional->SetupDerivedInstruction(else_call_op); in DoConditionalToSelect()
H A Dgather_expander.cc344 gather_instr->SetupDerivedInstruction(broadcast_operand); in ExpandInstruction()
349 gather_instr->SetupDerivedInstruction(broadcast); in ExpandInstruction()
H A Dconvolution_group_converter.cc313 convolution->SetupDerivedInstruction(new_convolution); in HandleBatchGroupCount()
668 convolution->SetupDerivedInstruction(new_convolution); in HandleConvolution()
H A Ddot_decomposer.cc172 original_dot->SetupDerivedInstruction(dot); in CanonicalizeDot()
H A Dhlo_creation_utils.cc503 derived_from->SetupDerivedInstruction(pred); in MakeSelectHlo()
516 derived_from->SetupDerivedInstruction(select); in MakeSelectHlo()
H A Dspace_to_batch_converter.cc2437 old_instr->SetupDerivedInstruction(output_transpose); in BatchToSpace()
2720 convolution->SetupDerivedInstruction(new_conv); in PropagateOnConv()
3425 convolution->SetupDerivedInstruction(new_conv); in PropagateOnBackpropFilterConv()
3730 convolution->SetupDerivedInstruction(batch_increased_reshape); in PerformSpaceToBatchOnConvolution()
3792 convolution->SetupDerivedInstruction(new_conv); in PerformSpaceToBatchOnConvolution()
H A Dconditional_simplifier.cc463 conditional->SetupDerivedInstruction(call); in TryRemoveConditional()
H A Dhlo_instruction.cc75 SetupDerivedInstruction(derived); in AddInstruction()
1805 void HloInstruction::SetupDerivedInstruction( in SetupDerivedInstruction() function in xla::HloInstruction
2159 SetupDerivedInstruction(clone.get()); in CloneWithNewOperands()
H A Dalgebraic_simplifier.cc4751 concat->SetupDerivedInstruction(new_concat); in HandleSlice()
5550 dot->SetupDerivedInstruction(new_dot); in HandleReduce()
5557 reduce->SetupDerivedInstruction(new_reduce); in HandleReduce()
6536 convolution->SetupDerivedInstruction(new_convolution); in SwapConvOperands()
H A Dhlo_instruction.h1597 void SetupDerivedInstruction(HloInstruction* derived_instruction) const;
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dcanonicalize_all_gather_for_cse.cc86 ag->SetupDerivedInstruction(new_ag); in RunOnComputation()