Searched refs:HloPassPipeline (Results 1 – 9 of 9) sorted by relevance
35 Status HloPassPipeline::RunInvariantCheckers( in RunInvariantCheckers()55 StatusOr<bool> HloPassPipeline::RunPassesInternal( in RunPassesInternal()76 std::vector<HloPassInterface*> HloPassPipeline::GetEnabledPasses( in GetEnabledPasses()100 void HloPassPipeline::MaybeDumpHlo(const HloModule& module, in MaybeDumpHlo()107 void HloPassPipeline::MaybeDumpHlo(const HloModuleGroup& module_group, in MaybeDumpHlo()115 StatusOr<bool> HloPassPipeline::Run(HloModule* module) { in Run()125 StatusOr<bool> HloPassPipeline::RunOnModuleGroup(HloModuleGroup* module_group) { in RunOnModuleGroup()
111 HloPassPipeline pipeline(TestName()); in TEST_F()134 HloPassPipeline pipeline(TestName()); in TEST_F()165 HloPassPipeline pipeline(TestName()); in TEST_F()199 HloPassPipeline pipeline(TestName()); in TEST_F()209 HloPassPipeline pipeline(TestName()); in TEST_F()223 HloPassPipeline pipeline(TestName()); in TEST_F()248 HloPassPipeline pipeline(TestName()); in TEST_F()
35 class HloPassPipeline : public HloPassInterface {37 explicit HloPassPipeline(const string& name) : name_(name) {} in HloPassPipeline() function
40 HloPassPipeline pipeline_;
85 static bool AddPasses(HloPassPipeline* pipeline,
286 bool ReducePrecisionInsertion::AddPasses(HloPassPipeline* pipeline, in AddPasses()
183 HloPassPipeline pipeline("optimization"); in OptimizeHloModule()211 pipeline.AddPass<HloPassFix<HloPassPipeline>>("simplification"); in OptimizeHloModule()270 HloPassPipeline pipeline("conv_canonicalization"); in OptimizeHloModule()297 HloPassPipeline pipeline("layout assignment"); in OptimizeHloModule()305 HloPassPipeline pipeline("post-layout_assignment"); in OptimizeHloModule()355 HloPassFix<HloPassPipeline> fusion("fusion"); in OptimizeHloModule()374 HloPassPipeline reduce_pipeline("reduce-precision"); in OptimizeHloModule()404 HloPassPipeline pipeline("GPU-ir-emit-prepare"); in PrepareHloModuleForIrEmitting()
250 HloPassPipeline pipeline("HLO passes through layout assignment"); in RunHloPassesThroughLayoutAssn()283 pipeline.AddPass<HloPassFix<HloPassPipeline>>("simplification"); in RunHloPassesThroughLayoutAssn()340 HloPassPipeline pipeline("HLO passes after layout assignment"); in RunHloPassesAfterLayoutAssn()343 pipeline.AddPass<HloPassPipeline>("after layout assignment"); in RunHloPassesAfterLayoutAssn()351 auto& pass = pipeline.AddPass<HloPassFix<HloPassPipeline>>( in RunHloPassesAfterLayoutAssn()
81 HloPassPipeline pipeline("Interpreter"); in RunHloOptimization()