Home
last modified time | relevance | path

Searched refs:CreateDynamicSlice (Results 1 – 20 of 20) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_instruction_fusion_test.cc334 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
378 HloInstruction::CreateDynamicSlice(result_shape, param0, {param1}, {2})); in TEST_F()
548 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
583 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
592 auto slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner_util.cc181 auto index = b->AddInstruction(HloInstruction::CreateDynamicSlice( in MakePartitionOffsets()
411 result = b->AddInstruction(HloInstruction::CreateDynamicSlice( in TileToPartialReplicateHaloExchange()
520 result = b->AddInstruction(HloInstruction::CreateDynamicSlice( in PadFromPartialReplicateShape()
969 valid_slice = b->AddInstruction(HloInstruction::CreateDynamicSlice( in ExchangeHaloAndGetValidData()
1472 b->AddInstruction(HloInstruction::CreateDynamicSlice( in GetInGroupPartitionId()
1595 b->AddInstruction(HloInstruction::CreateDynamicSlice( in PerGroupSliceFromReplicated()
1615 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PerGroupSliceFromReplicated()
H A Dspmd_partitioner.cc559 auto slice = state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardNoCache()
993 state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardAsWindowedInput()
1437 auto slice = state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardFromPartialReplicateWithDynamicSlice()
2378 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleConcatenate()
3017 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
3314 branch_index = b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleInfeed()
3448 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandlePad()
3743 branch_index = b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleOutfeed()
3947 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleReduceWindow()
4169 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleSelectAndScatter()
H A Dconvolution_handler.cc457 conv_lhs = b->AddInstruction(HloInstruction::CreateDynamicSlice( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
811 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PartitionConvolutionTiledOutput()
H A Dfft_handler.cc104 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PadEachPartitionWithHaloExchange()
H A Ddot_handler.cc4234 return body->AddInstruction(HloInstruction::CreateDynamicSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
4395 body->AddInstruction(HloInstruction::CreateDynamicSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dreduce_scatter_decomposer.cc76 computation->AddInstruction(HloInstruction::CreateDynamicSlice( in Run()
H A Ddynamic_index_splitter.cc91 : HloInstruction::CreateDynamicSlice( in Run()
H A Dcollective_decomposer_utils.cc143 computation->AddInstruction(HloInstruction::CreateDynamicSlice( in CreateStartIndicesForCollectiveDecomposition()
H A Ddynamic_padder.cc889 reverse->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteReverse()
955 input = conv->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteInputWithDynamicPadding()
1287 rewritten = hlo->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteDynamicSelectAndScatterSamePadding()
1620 dus->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteDynamicUpdateSlice()
H A Dhlo_creation_utils.cc173 HloInstruction::CreateDynamicSlice(dynamic_slice_shape, operand, in MakeDynamicSliceHlo()
202 HloInstruction::CreateDynamicSlice(dynamic_slice_shape, operand, in MakeDynamicSliceHlo()
H A Ddynamic_dimension_inference_test.cc1111 auto* slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
1213 auto* slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
H A Dalgebraic_simplifier.cc757 auto sliced_lhs = lhs->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleAdd()
2388 dot->AddInstruction(HloInstruction::CreateDynamicSlice( in OptimizeDotOfGather()
4840 dynamic_slice->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
4875 dynamic_slice->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
4889 dynamic_slice->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
H A Dhlo_instruction.h889 static std::unique_ptr<HloInstruction> CreateDynamicSlice(
H A Dalgebraic_simplifier_test.cc5001 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
5025 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
5059 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
6316 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
6397 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
H A Dhlo_evaluator_test.cc3063 HloInstruction::CreateDynamicSlice(shape, operand, {zero, one}, {2, 3})); in TEST_P()
3101 HloInstruction::CreateDynamicSlice(shape, operand, {two, one}, {2, 3})); in TEST_P()
H A Dhlo_instruction.cc813 instruction = CreateDynamicSlice( in CreateFromProto()
1513 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateDynamicSlice( in CreateDynamicSlice() function in xla::HloInstruction
H A Dhlo_dataflow_analysis_test.cc2304 auto ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
H A Dhlo_parser.cc2293 return builder->AddInstruction(HloInstruction::CreateDynamicSlice( in CreateInstruction()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/tests/
H A Dcpu_gpu_fusion_test.cc541 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in XLA_TEST_F()