Searched defs:dus (Results 1 – 8 of 8) sorted by relevance
127 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather() local
2747 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local2787 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local
2315 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local2485 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local
1180 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F() local
1288 HloDynamicUpdateSliceInstruction* dus = in RewriteDynamicUpdateSlice() local
795 auto dus = mlir::dyn_cast<mlir::mhlo::DynamicUpdateSliceOp>( in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
890 auto dus = in ReplicatePartial() local2440 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local
3214 auto dus = body_root->mutable_operand(2); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local