Searched refs:horizontal_fusion (Results 1 – 1 of 1) sorted by relevance
329 HloPassPipeline horizontal_fusion("horizontal_fusion"); in OptimizeHloModule() local330 horizontal_fusion.AddPass<GpuHorizontalLoopFusion>(); in OptimizeHloModule()331 horizontal_fusion.AddPass<GpuHorizontalInputFusion>(); in OptimizeHloModule()332 horizontal_fusion.AddPass<HloCSE>(/*is_layout_sensitive=*/true, in OptimizeHloModule()334 horizontal_fusion.AddPass<HloDCE>(); in OptimizeHloModule()335 TF_RETURN_IF_ERROR(horizontal_fusion.Run(hlo_module).status()); in OptimizeHloModule()