Searched refs:func_pm (Results 1 – 5 of 5) sorted by relevance
158 OpPassManager &func_pm = pm.nest<func::FuncOp>(); in CreateTFStandardPipeline() local164 func_pm.addPass(tf_executor::CreateTFExecutorGraphPruningPass()); in CreateTFStandardPipeline()165 func_pm.addPass(tf_executor::CreateTFExecutorIslandCoarseningPass()); in CreateTFStandardPipeline()166 func_pm.addPass(CreateMaterializePassthroughOpPass()); in CreateTFStandardPipeline()168 func_pm.addPass(TFDevice::CreateClusterFormationPass()); in CreateTFStandardPipeline()172 func_pm.addPass(createCanonicalizerPass()); in CreateTFStandardPipeline()
120 OpPassManager &func_pm = pm.nest<func::FuncOp>(); in CreateTPUBridgePipelineImpl() local121 func_pm.addPass(CreateTPUHostComputationExpansionPass()); in CreateTPUBridgePipelineImpl()122 func_pm.addPass(CreateTPUUpdateEmbeddingEnqueueOpInputsPass()); in CreateTPUBridgePipelineImpl()
194 mlir::OpPassManager& func_pm = pm_.nest<mlir::func::FuncOp>(); in TFRDecomposeContext() local197 func_pm.addPass(mlir::CreateExecutorDialectToFunctionalConversionPass()); in TFRDecomposeContext()200 func_pm.addPass(mlir::TFR::CreateDecomposeTFOpsPass(tfr_module_)); in TFRDecomposeContext()201 func_pm.addPass(mlir::TFR::CreateRaiseToTFOpsPass( in TFRDecomposeContext()205 func_pm.addPass(mlir::CreateFunctionalToExecutorDialectConversionPass()); in TFRDecomposeContext()
128 mlir::OpPassManager &func_pm = pm->nest<mlir::func::FuncOp>(); in CreateDTensorMLIRPass() local129 func_pm.addPass(CreateDTensorDeviceMeshClusterCoarsening()); in CreateDTensorMLIRPass()132 func_pm.addPass(CreateDTensorDesignateResourceHandleMesh()); in CreateDTensorMLIRPass()
413 OpPassManager& func_pm = pm.nest<func::FuncOp>(); in CreateTFLStandardPipeline() local417 mlir::TF::CreateTFStandardPipeline(func_pm, standard_pipeline_options); in CreateTFLStandardPipeline()