Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Ddynamic_index_splitter.cc85 ? HloInstruction::CreateDynamicUpdateSlice( in Run()
Dall_gather_decomposer.cc70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather()
Dhlo_dataflow_analysis_test.cc2210 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2253 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2393 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2491 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2532 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2563 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
Dtuple_points_to_analysis_test.cc711 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in Run()
941 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
Dhlo_creation_utils.cc201 return computation->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in MakeDynamicUpdateSliceHlo()
Dhlo_alias_analysis_test.cc1079 HloInstruction::CreateDynamicUpdateSlice(shape, copy0, param1, {param2})); in TEST_F()
Dhlo_instruction.h890 static std::unique_ptr<HloInstruction> CreateDynamicUpdateSlice(
Dhlo_instruction.cc775 CreateDynamicUpdateSlice(shape, operands(0), operands(1), in CreateFromProto()
1424 HloInstruction::CreateDynamicUpdateSlice( in CreateDynamicUpdateSlice() function in xla::HloInstruction
1977 clone = CreateDynamicUpdateSlice(shape, new_operands[0], new_operands[1], in CloneWithNewOperands()
Ddynamic_padder.cc1268 comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in RewriteDynamicConcat()
Dhlo_evaluator_test.cc3101 b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_P()
Dalgebraic_simplifier_test.cc4686 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
5528 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
Dhlo_parser.cc2086 builder->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in ParseInstructionRhs()
Dalgebraic_simplifier.cc910 auto dynamic_update_slice_full = HloInstruction::CreateDynamicUpdateSlice( in HandleAdd()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_instruction_fusion_test.cc573 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
617 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Ddot_handler.cc1140 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
1142 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
1209 o = body_b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in PartitionBaseCase()
3734 body->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
Dspmd_partitioner.cc941 state_.b->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in ReplicatePartial()
1660 temp_output = b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleConcatenate()
2414 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice()
2443 return b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice()