Home
last modified time | relevance | path

Searched defs:dus (Results 1 – 14 of 14) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dinstruction_fusion.cc860 const HloInstruction* dus = in IsSafeToFuseSliceIntoDusFusion() local
945 const HloInstruction* dus = in ShouldFuseInPlaceOp() local
H A Dall_gather_decomposer.cc70 auto dus = comp->AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in DecomposeAllGather() local
H A Dhlo_dataflow_analysis_test.cc2307 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local
2484 auto dus = builder.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in TEST_F() local
3022 HloInstruction* dus = module->entry_computation()->root_instruction(); in TEST_F() local
H A Dsharding_propagation_test.cc3146 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local
3186 auto* dus = FindInstruction(module.get(), "dus"); in TEST_P() local
H A Dhlo_cost_analysis_test.cc676 auto dus = dus_module->entry_computation()->root_instruction(); in TEST_F() local
H A Dbfloat16_propagation_test.cc1137 HloInstruction* dus = module->entry_computation()->GetInstructionWithName( in TEST_F() local
H A Ddynamic_padder.cc1547 HloDynamicUpdateSliceInstruction* dus = in RewriteDynamicUpdateSlice() local
H A Dcopy_insertion_test.cc3066 auto dus = add1->operand(0); in TEST_F() local
H A Dalgebraic_simplifier.cc4343 HloInstruction* dus; in HandleReshape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dir_emission_utils.cc632 auto dus = mlir::dyn_cast<mlir::mhlo::DynamicUpdateSliceOp>( in CanEmitFusedDynamicUpdateSliceInPlaceForGpu() local
H A Dgpu_fusible_test.cc926 const HloInstruction* dus = tuple->operand(0); in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc4177 HloInstruction* dus; in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() member
4188 HloInstruction* dus = body_root->mutable_operand(2); in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions() local
4233 HloInstruction* dus) { in MoveUsersIntoWindowedDotGeneralLoopOnNonContractingDimensions()
H A Dspmd_partitioner.cc1286 auto dus = in ReplicatePartial() local
3070 auto dus = b_.AddInstruction(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local
3191 auto dus = add_hlo(HloInstruction::CreateDynamicUpdateSlice( in HandleDynamicUpdateSlice() local
H A Dspmd_partitioner_test.cc10737 auto dus = in TEST_F() local