Searched defs:dus (Results 1 – 10 of 10) sorted by relevance
70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather() local
2896 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local2936 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local
2393 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local2563 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local
1180 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F() local
1480 HloDynamicUpdateSliceInstruction* dus = in RewriteDynamicUpdateSlice() local
3087 auto dus = add1->operand(0); in TEST_F() local
4066 HloInstruction* dus; in HandleReshape() local
891 auto dus = mlir::dyn_cast<mlir::mhlo::DynamicUpdateSliceOp>( in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
940 auto dus = in ReplicatePartial() local2414 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local
3587 auto dus = body_root->mutable_operand(2); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local