Searched refs:HloPassFix (Results 1 – 5 of 5) sorted by relevance
32 class HloPassFix : public Pass {35 explicit HloPassFix(Args&&... args) : Pass(args...) {} in HloPassFix() function
1739 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()1783 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()1826 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()1855 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()1881 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()2802 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()2823 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()2846 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()2871 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()2893 HloPassFix<AlgebraicSimplifier> simplifier(default_options_); in TEST_F()
220 pipeline.AddPass<HloPassFix<HloPassPipeline>>("simplification"); in OptimizeHloModule()310 HloPassFix<HloPassPipeline> fusion("fusion"); in OptimizeHloModule()416 pipeline.AddPass<HloPassFix<ReductionSplitter>>(); in OptimizeHloPostLayoutAssignment()432 pipeline.AddPass<HloPassFix<AlgebraicSimplifier>>(options); in OptimizeHloPostLayoutAssignment()437 pipeline.AddPass<HloPassFix<GpuTreeReductionRewriter>>(); in OptimizeHloPostLayoutAssignment()
130 auto& pass = pipeline.AddPass<HloPassFix<HloPassPipeline>>( in OptimizeHloConvolutionCanonicalization()
345 pipeline.AddPass<HloPassFix<HloPassPipeline>>("simplification"); in RunHloPassesThroughLayoutAssn()415 auto& pass = pipeline.AddPass<HloPassFix<HloPassPipeline>>( in RunHloPassesAfterLayoutAssn()424 pass.AddPass<HloPassFix<AlgebraicSimplifier>>(options); in RunHloPassesAfterLayoutAssn()