Searched refs:CreateDynamicUpdateSlice (Results 1 – 16 of 16) sorted by relevance
85 ? HloInstruction::CreateDynamicUpdateSlice( in Run()
70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather()
2210 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()
711 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in Run()941 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
201 return computation->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in MakeDynamicUpdateSliceHlo()
1079 HloInstruction::CreateDynamicUpdateSlice(shape, copy0, param1, {param2})); in TEST_F()
890 static std::unique_ptr<HloInstruction> CreateDynamicUpdateSlice(
775 CreateDynamicUpdateSlice(shape, operands(0), operands(1), in CreateFromProto()1424 HloInstruction::CreateDynamicUpdateSlice( in CreateDynamicUpdateSlice() function in xla::HloInstruction1977 clone = CreateDynamicUpdateSlice(shape, new_operands[0], new_operands[1], in CloneWithNewOperands()
1268 comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in RewriteDynamicConcat()
3101 b.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_P()
4686 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()5528 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
2086 builder->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in ParseInstructionRhs()
910 auto dynamic_update_slice_full = HloInstruction::CreateDynamicUpdateSlice( in HandleAdd()
573 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()617 builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F()
1140 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()
941 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()