Searched refs:post_pipeline (Results 1 – 1 of 1) sorted by relevance
175 HloPassPipeline post_pipeline("nvptx post-layout_assignment part 2"); in OptimizeHloPostLayoutAssignment() local178 post_pipeline.AddPass<GemmAlgorithmPicker>(stream_exec, device_allocator); in OptimizeHloPostLayoutAssignment()179 TF_RETURN_IF_ERROR(post_pipeline.Run(hlo_module).status()); in OptimizeHloPostLayoutAssignment()