Home
last modified time | relevance | path

Searched refs:HloDCE (Results 1 – 21 of 21) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_dce.h36 class HloDCE : public HloModulePass {
38 HloDCE() : remove_cross_partition_collective_ops_(false) {} in HloDCE() function
39 explicit HloDCE(bool remove_cross_partition_collective_ops) in HloDCE() function
42 ~HloDCE() override {} in ~HloDCE()
Dhlo_dce_test.cc67 HloDCE dce; in TEST_F()
89 HloDCE dce; in TEST_F()
108 HloDCE dce; in TEST_F()
125 HloDCE dce; in TEST_F()
155 HloDCE dce; in TEST_F()
202 HloDCE dce; in TEST_F()
243 HloDCE dce; in TEST_F()
301 HloDCE dce; in TEST_F()
353 HloDCE dce; in TEST_F()
395 HloDCE dce; in TEST_F()
[all …]
Dtopk_rewriter_test.cc82 TF_ASSERT_OK(HloDCE().Run(module.get()).status()); in TEST_F()
112 TF_ASSERT_OK(HloDCE().Run(module.get()).status()); in TEST_F()
142 TF_ASSERT_OK(HloDCE().Run(module.get()).status()); in TEST_F()
Dhlo_schedule_test.cc156 HloDCE dce; in TEST_F()
198 HloDCE dce; in TEST_F()
266 HloDCE dce; in TEST_F()
329 HloDCE dce; in TEST_F()
Dhlo_dce.cc40 StatusOr<bool> HloDCE::RunOnComputation( in RunOnComputation()
76 StatusOr<bool> HloDCE::Run(HloModule* module) { in Run()
Dhlo_module_dce.cc97 HloDCE().RunOnComputation( in RunWhileDCE()
130 HloDCE hlo_dce; in Run()
Dwhile_loop_simplifier_test.cc607 EXPECT_TRUE(HloDCE().Run(m.get()).ok()); in TEST_F()
649 EXPECT_TRUE(HloDCE().Run(m.get()).ok()); in TEST_F()
691 EXPECT_TRUE(HloDCE().Run(m.get()).ok()); in TEST_F()
759 EXPECT_TRUE(HloDCE().Run(m.get()).ok()); in TEST_F()
Dcall_inliner.cc163 TF_RETURN_IF_ERROR(HloDCE().Run(module).status()); in Run()
Dconditional_code_motion.cc1288 subpipeline.AddPass<HloDCE>(); in Run()
1508 subpipeline.AddPass<HloDCE>(); in Run()
1510 subpipeline.AddPass<HloDCE>(); in Run()
Dmulti_output_fusion.cc132 HloDCE dce; in Run()
Dbfloat16_normalization.cc536 HloDCE dce; in Run()
Dbfloat16_propagation.cc965 HloDCE dce; in Run()
Ddynamic_padder_test.cc391 HloDCE dce; in PadAndExecute()
Dcopy_insertion.cc1256 HloDCE dce; in Run()
Dhlo_rematerialization.cc2074 TF_ASSIGN_OR_RETURN(bool dead_code_removed, HloDCE().Run(module)); in Run()
Ddynamic_padder.cc1959 HloDCE dce; in Run()
Dlayout_assignment.cc2361 HloDCE dce; in ClearPreviousPassSideEffects()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dhorizontal_loop_fusion_test.cc74 EXPECT_TRUE(HloDCE().Run(module.get()).ValueOrDie()); in TEST_F()
284 EXPECT_TRUE(HloDCE().Run(module.get()).ValueOrDie()); in TEST_F()
Dgpu_compiler.cc254 pass.AddPass<HloDCE>(); in OptimizeHloModule()
268 pipeline.AddPass<HloDCE>(); in OptimizeHloModule()
326 fusion.AddPass<HloDCE>(); in OptimizeHloModule()
334 horizontal_fusion.AddPass<HloDCE>(); in OptimizeHloModule()
379 pipeline.AddPass<HloDCE>(); in PrepareHloModuleForIrEmitting()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc354 pass.AddPass<HloDCE>(); in RunHloPassesThroughLayoutAssn()
369 pass.AddPass<HloDCE>(); in RunHloPassesThroughLayoutAssn()
425 pass.AddPass<HloDCE>(); in RunHloPassesAfterLayoutAssn()
449 pipeline.AddPass<HloDCE>(); in RunHloPassesAfterLayoutAssn()
451 pipeline.AddPass<HloDCE>(); in RunHloPassesAfterLayoutAssn()
/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3723 pass.AddPass<HloDCE>(); in Run()