Searched refs:pre_pipeline (Results 1 – 1 of 1) sorted by relevance
163 HloPassPipeline pre_pipeline("nvptx post-layout_assignment part 1"); in OptimizeHloPostLayoutAssignment() local168 pre_pipeline.AddPass<CublasGemmPadForTensorCores>(); in OptimizeHloPostLayoutAssignment()170 TF_RETURN_IF_ERROR(pre_pipeline.Run(hlo_module).status()); in OptimizeHloPostLayoutAssignment()