Searched refs:func_pm (Results 1 – 4 of 4) sorted by relevance
148 OpPassManager &func_pm = pm.nest<FuncOp>(); in CreateTFStandardPipeline() local154 func_pm.addPass(tf_executor::CreateTFExecutorGraphPruningPass()); in CreateTFStandardPipeline()155 func_pm.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass()); in CreateTFStandardPipeline()156 func_pm.addPass(CreateMaterializePassthroughOpPass()); in CreateTFStandardPipeline()158 func_pm.addPass(TFDevice::CreateClusterFormationPass()); in CreateTFStandardPipeline()162 func_pm.addPass(createCanonicalizerPass()); in CreateTFStandardPipeline()
105 OpPassManager &func_pm = pm.nest<FuncOp>(); in CreateTPUBridgePipeline() local109 func_pm.addPass(TFDevice::CreateDecomposeResourceOpsPass()); in CreateTPUBridgePipeline()110 func_pm.addPass(CreateTPUHostComputationExpansionPass()); in CreateTPUBridgePipeline()111 func_pm.addPass(CreateTPUUpdateEmbeddingEnqueueOpInputsPass()); in CreateTPUBridgePipeline()
192 mlir::OpPassManager& func_pm = pm_.nest<mlir::FuncOp>(); in TFRDecomposeContext() local195 func_pm.addPass(mlir::CreateExecutorDialectToFunctionalConversionPass()); in TFRDecomposeContext()198 func_pm.addPass(mlir::TFR::CreateDecomposeTFOpsPass(tfr_module_)); in TFRDecomposeContext()199 func_pm.addPass(mlir::TFR::CreateRaiseToTFOpsPass( in TFRDecomposeContext()203 func_pm.addPass(mlir::CreateFunctionalToExecutorDialectConversionPass()); in TFRDecomposeContext()
274 OpPassManager& func_pm = pm.nest<FuncOp>(); in CreateTFLStandardPipeline() local278 mlir::TF::CreateTFStandardPipeline(func_pm, standard_pipeline_options); in CreateTFLStandardPipeline()