Home
last modified time | relevance | path

Searched refs:CreateDynamicUpdateSlice (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Ddynamic_index_splitter.cc88 ? HloInstruction::CreateDynamicUpdateSlice( in Run()
H A Dall_gather_decomposer.cc70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather()
H A Dhlo_dataflow_analysis_test.cc2117 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2161 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2307 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2410 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2452 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2484 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
H A Dhlo_creation_utils.cc232 HloInstruction::CreateDynamicUpdateSlice( in MakeDynamicUpdateSliceHlo()
H A Dhlo_alias_analysis_test.cc913 HloInstruction::CreateDynamicUpdateSlice(shape, copy0, param1, {param2})); in TEST_F()
H A Dhlo_instruction.cc833 CreateDynamicUpdateSlice(shape, operands(0), operands(1), in CreateFromProto()
1522 HloInstruction::CreateDynamicUpdateSlice( in CreateDynamicUpdateSlice() function in xla::HloInstruction
2120 clone = CreateDynamicUpdateSlice(shape, new_operands[0], new_operands[1], in CloneWithNewOperands()
H A Dhlo_instruction.h896 static std::unique_ptr<HloInstruction> CreateDynamicUpdateSlice(
H A Ddynamic_padder.cc1317 concat->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in RewriteDynamicConcat()
H A Dhlo_evaluator_test.cc3139 b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_P()
H A Dalgebraic_simplifier_test.cc5066 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
6136 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
H A Dhlo_parser.cc2312 return builder->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in CreateInstruction()
H A Dalgebraic_simplifier.cc765 auto dynamic_update_slice_full = HloInstruction::CreateDynamicUpdateSlice( in HandleAdd()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc555 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
599 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.cc1287 state_.b->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in ReplicatePartial()
2352 temp_output = b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleConcatenate()
3070 auto dus = b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice()
3191 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice()
3224 return b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice()
H A Ddot_handler.cc1402 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
1404 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
1471 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
4407 body->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()