Searched defs:dus (Results 1 – 14 of 14) sorted by relevance
860 const HloInstruction* dus = in IsSafeToFuseSliceIntoDusFusion() local945 const HloInstruction* dus = in ShouldFuseInPlaceOp() local
70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather() local
2307 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local2484 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local3022 HloInstruction* dus = module->entry_computation()->root_instruction(); in TEST_F() local
3146 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local3186 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local
676 auto dus = dus_module->entry_computation()->root_instruction(); in TEST_F() local
1137 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F() local
1547 HloDynamicUpdateSliceInstruction* dus = in RewriteDynamicUpdateSlice() local
3066 auto dus = add1->operand(0); in TEST_F() local
4343 HloInstruction* dus; in HandleReshape() local
632 auto dus = mlir::dyn_cast<mlir::mhlo::DynamicUpdateSliceOp>( in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
926 const HloInstruction* dus = tuple->operand(0); in TEST_F() local
4177 HloInstruction* dus; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() member4188 HloInstruction* dus = body_root->mutable_operand(2); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local4233 HloInstruction* dus) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
1286 auto dus = in ReplicatePartial() local3070 auto dus = b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local3191 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local
10737 auto dus = in TEST_F() local