Searched refs:addNestedPass (Results 1 – 17 of 17) sorted by relevance
73 pm.addNestedPass<mlir::FuncOp>(::mlir::createBufferHoistingPass()); in LowerLHLOToGPU()74 pm.addNestedPass<mlir::FuncOp>(::mlir::createBufferDeallocationPass()); in LowerLHLOToGPU()76 pm.addNestedPass<mlir::FuncOp>(createFusionOpRemoverPass()); in LowerLHLOToGPU()78 pm.addNestedPass<mlir::FuncOp>(::mlir::createCopyRemovalPass()); in LowerLHLOToGPU()80 pm.addNestedPass<mlir::FuncOp>(::mlir::lmhlo::createLegalizeToGpuPass()); in LowerLHLOToGPU()82 pm.addNestedPass<mlir::FuncOp>( in LowerLHLOToGPU()85 pm.addNestedPass<mlir::FuncOp>(::mlir::lmhlo::createLhloFuseLinalgPass( in LowerLHLOToGPU()89 pm.addNestedPass<mlir::FuncOp>( in LowerLHLOToGPU()93 pm.addNestedPass<::mlir::FuncOp>(::mlir::createCanonicalizerPass()); in LowerLHLOToGPU()94 pm.addNestedPass<::mlir::FuncOp>(::mlir::createCSEPass()); in LowerLHLOToGPU()[all …]
47 pass_manager->addNestedPass<mlir::FuncOp>( in AddQuantizationPasses()51 pass_manager->addNestedPass<mlir::FuncOp>( in AddQuantizationPasses()57 pass_manager->addNestedPass<mlir::FuncOp>( in AddQuantizationPasses()61 pass_manager->addNestedPass<mlir::FuncOp>( in AddQuantizationPasses()72 pass_manager->addNestedPass<mlir::FuncOp>( in AddTFToTFLConversionPasses()116 pass_manager->addNestedPass<mlir::FuncOp>( in AddTFToTFLConversionPasses()161 pass_manager->addNestedPass<mlir::FuncOp>( in AddTFToTFLConversionPasses()166 pass_manager->addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in AddTFToTFLConversionPasses()167 pass_manager->addNestedPass<mlir::FuncOp>(mlir::createCSEPass()); in AddTFToTFLConversionPasses()194 pass_manager->addNestedPass<mlir::FuncOp>(mlir::TFL::CreatePrepareTFPass( in AddTFToTFLConversionPasses()[all …]
202 pm.addNestedPass<mlir::FuncOp>(mlir::mhlo::createLegalizeTFPass( in LowerTFtoLoops()204 pm.addNestedPass<mlir::FuncOp>(mlir::createTransformUnrankedHloPass()); in LowerTFtoLoops()205 pm.addNestedPass<mlir::FuncOp>(mlir::mhlo::createChloLegalizeToHloPass()); in LowerTFtoLoops()206 pm.addNestedPass<mlir::FuncOp>(mlir::mhlo::createLowerComplexPass()); in LowerTFtoLoops()207 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in LowerTFtoLoops()208 pm.addNestedPass<mlir::FuncOp>(mlir::createCSEPass()); in LowerTFtoLoops()209 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in LowerTFtoLoops()212 pm.addNestedPass<mlir::FuncOp>(::mlir::mhlo::createLegalizeHloToLinalgPass()); in LowerTFtoLoops()214 pm.addNestedPass<mlir::FuncOp>(mlir::createCSEPass()); in LowerTFtoLoops()216 pm.addNestedPass<mlir::FuncOp>(mlir::createLinalgFusionOfTensorOpsPass()); in LowerTFtoLoops()[all …]
45 pm.addNestedPass<FuncOp>(std::move(pass)); in AddGraphExportLoweringPasses()53 pm.addNestedPass<FuncOp>(CreateTPUDevicePropagationPass()); in AddGraphExportLoweringPasses()91 pm.addNestedPass<FuncOp>( in CreateTPUBridgePipeline()96 pm.addNestedPass<FuncOp>(CreateExecutorDialectToFunctionalConversionPass()); in CreateTPUBridgePipeline()100 pm.addNestedPass<FuncOp>(CreateTPUReorderReplicateAndPartitionedInputsPass()); in CreateTPUBridgePipeline()120 pm.addNestedPass<FuncOp>(createCSEPass()); in CreateTPUBridgePipeline()125 pm.addNestedPass<FuncOp>(TFDevice::CreateClusterConstantSinkingPass()); in CreateTPUBridgePipeline()131 pm.addNestedPass<FuncOp>(CreateTPUResourceReadsWritesPartitioningPass()); in CreateTPUBridgePipeline()135 pm.addNestedPass<FuncOp>(TFDevice::CreateReplicateInvariantOpHoistingPass()); in CreateTPUBridgePipeline()136 pm.addNestedPass<FuncOp>(CreateTPUMergeVariablesWithExecutePass()); in CreateTPUBridgePipeline()[all …]
168 pm.addNestedPass<FuncOp>(CreateTFOptimizePass()); in CreateTFStandardPipeline()169 pm.addNestedPass<FuncOp>(createCSEPass()); in CreateTFStandardPipeline()
49 pm.addNestedPass<FuncOp>(tf_executor::CreateTFExecutorIslandCoarseningPass()); in Run()
88 pm.addNestedPass<FuncOp>(CreateInitTextFileToImportPass()); in runOnOperation()
39 pm->addNestedPass<mlir::FuncOp>(mlir::tfjs::CreateOptimizePass()); in AddTFToTFJSConversionPasses()42 pm->addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in AddTFToTFJSConversionPasses()43 pm->addNestedPass<mlir::FuncOp>(mlir::createCSEPass()); in AddTFToTFJSConversionPasses()46 pm->addNestedPass<mlir::FuncOp>( in AddTFToTFJSConversionPasses()
283 pm.addNestedPass<mlir::FuncOp>(mlir::TF::CreateDropWhileShapeInvariantPass()); in CreateConvertMlirToXlaHloPipeline()284 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in CreateConvertMlirToXlaHloPipeline()296 pm.addNestedPass<mlir::FuncOp>( in CreateConvertMlirToXlaHloPipeline()311 pm.addNestedPass<mlir::FuncOp>(mlir::mhlo::createLegalizeTFPass( in CreateConvertMlirToXlaHloPipeline()315 pm.addNestedPass<mlir::FuncOp>(std::move(target_pass)); in CreateConvertMlirToXlaHloPipeline()318 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in CreateConvertMlirToXlaHloPipeline()328 pm.addNestedPass<mlir::FuncOp>(mlir::mhlo::createLegalizeTFPass( in CreateConvertMlirToXlaHloPipeline()337 pm.addNestedPass<mlir::FuncOp>( in CreateConvertMlirToXlaHloPipeline()
103 pm.addNestedPass<FuncOp>(createConvertLinalgToLoopsPass()); in runOnOperation()
63 pm.addNestedPass<FuncOp>(std::make_unique<AnnotateFunctionPass>()); in TEST()
121 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass()); in dumpMLIR()
87 template <typename OpT> void addNestedPass(std::unique_ptr<Pass> pass) { in addNestedPass() function
128 pm.addNestedPass<mlir::FuncOp>(mlir::createCanonicalizerPass());
526 pm.addNestedPass<FuncOp>(CreateFunctionalToExecutorDialectConversionPass()); in GetFunctionDef()
3840 pm.addNestedPass<mlir::FuncOp>( in LiftVariables()3842 pm.addNestedPass<mlir::FuncOp>( in LiftVariables()3846 pm.addNestedPass<mlir::FuncOp>( in LiftVariables()3852 pm.addNestedPass<mlir::FuncOp>( in LiftVariables()
1773 pm.addNestedPass<mlir::FuncOp>(mhlo::CreatePrepareForExport()); in ConvertMlirHloToHlo()