Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dreduce_scatter_decomposer.cc72 computation->AddInstruction(HloInstruction::CreateDynamicSlice( in Run()
Ddynamic_index_splitter.cc88 : HloInstruction::CreateDynamicSlice( in Run()
Dcollective_decomposer_utils.cc135 computation->AddInstruction(HloInstruction::CreateDynamicSlice( in CreateStartIndicesForCollectiveDecomposition()
Dhlo_creation_utils.cc146 return computation->AddInstruction(HloInstruction::CreateDynamicSlice( in MakeDynamicSliceHlo()
173 return computation->AddInstruction(HloInstruction::CreateDynamicSlice( in MakeDynamicSliceHlo()
Ddynamic_padder.cc834 comp->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteReverse()
899 input = comp->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteInputWithDynamicPadding()
1237 rewritten = comp->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteDynamicSelectAndScatterSamePadding()
1554 comp->AddInstruction(HloInstruction::CreateDynamicSlice( in RewriteDynamicUpdateSlice()
Ddynamic_dimension_inference_test.cc1104 auto* slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
1206 auto* slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
Dhlo_instruction.h883 static std::unique_ptr<HloInstruction> CreateDynamicSlice(
Dalgebraic_simplifier_test.cc4621 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
4645 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
4679 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
5708 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
5789 auto* ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_P()
Dhlo_evaluator_test.cc3025 HloInstruction::CreateDynamicSlice(shape, operand, {zero, one}, {2, 3})); in TEST_P()
3063 HloInstruction::CreateDynamicSlice(shape, operand, {two, one}, {2, 3})); in TEST_P()
Dalgebraic_simplifier.cc902 computation_->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleAdd()
2143 computation_->AddInstruction(HloInstruction::CreateDynamicSlice( in OptimizeDotOfGather()
4523 computation->AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
Dhlo_instruction.cc755 instruction = CreateDynamicSlice( in CreateFromProto()
1415 /* static */ std::unique_ptr<HloInstruction> HloInstruction::CreateDynamicSlice( in CreateDynamicSlice() function in xla::HloInstruction
Dhlo_dataflow_analysis_test.cc2390 auto ds = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
Dhlo_parser.cc2067 instruction = builder->AddInstruction(HloInstruction::CreateDynamicSlice( in ParseInstructionRhs()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc352 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
396 HloInstruction::CreateDynamicSlice(result_shape, param0, {param1}, {2})); in TEST_F()
566 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
601 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
610 auto slice = builder.AddInstruction(HloInstruction::CreateDynamicSlice( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.cc230 auto index = b->AddInstruction(HloInstruction::CreateDynamicSlice( in MakePartitionOffsets()
496 result = b->AddInstruction(HloInstruction::CreateDynamicSlice( in TileToPartialReplicateHaloExchange()
604 result = b->AddInstruction(HloInstruction::CreateDynamicSlice( in PadFromPartialReplicateShape()
1053 valid_slice = b->AddInstruction(HloInstruction::CreateDynamicSlice( in ExchangeHaloAndGetValidData()
1633 b->AddInstruction(HloInstruction::CreateDynamicSlice( in GetInGroupPartitionId()
1755 b->AddInstruction(HloInstruction::CreateDynamicSlice( in PerGroupSliceFromReplicated()
1775 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PerGroupSliceFromReplicated()
Dspmd_partitioner.cc477 auto slice = state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardNoCache()
766 state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardAsWindowedInput()
1091 auto slice = state_.b->AddInstruction(HloInstruction::CreateDynamicSlice( in ReshardFromPartialReplicateWithDynamicSlice()
1685 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleConcatenate()
2271 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleDynamicSlice()
2533 branch_index = b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleInfeed()
2667 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandlePad()
2958 branch_index = b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleOutfeed()
3159 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleReduceWindow()
3381 return b_.AddInstruction(HloInstruction::CreateDynamicSlice( in HandleSelectAndScatter()
Dfft_handler.cc106 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PadEachPartitionWithHaloExchange()
Dconvolution_handler.cc457 conv_lhs = b->AddInstruction(HloInstruction::CreateDynamicSlice( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
809 return b->AddInstruction(HloInstruction::CreateDynamicSlice( in PartitionConvolutionTiledOutput()
Ddot_handler.cc3596 return body->AddInstruction(HloInstruction::CreateDynamicSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
3722 body->AddInstruction(HloInstruction::CreateDynamicSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcpu_gpu_fusion_test.cc543 builder.AddInstruction(HloInstruction::CreateDynamicSlice( in XLA_TEST_F()