Searched refs:TFDevice (Results 1 – 20 of 20) sorted by relevance
50 add_pass(TFDevice::CreateReplicateToIslandPass()); in AddGraphExportLoweringPasses()51 add_pass(TFDevice::CreateParallelExecuteToIslandsPass()); in AddGraphExportLoweringPasses()52 add_pass(TFDevice::CreateLaunchToDeviceAttributePass()); in AddGraphExportLoweringPasses()109 func_pm.addPass(TFDevice::CreateDecomposeResourceOpsPass()); in CreateTPUBridgePipeline()119 pm.addPass(TFDevice::CreateResourceOpLiftingPass()); in CreateTPUBridgePipeline()121 pm.addPass(TFDevice::CreateMarkOpsForOutsideCompilationPass()); in CreateTPUBridgePipeline()125 pm.addNestedPass<FuncOp>(TFDevice::CreateClusterConstantSinkingPass()); in CreateTPUBridgePipeline()127 pm.addPass(TFDevice::CreateClusterOutliningPass()); in CreateTPUBridgePipeline()132 pm.addPass(TFDevice::CreateAnnotateParameterReplicationPass()); in CreateTPUBridgePipeline()135 pm.addNestedPass<FuncOp>(TFDevice::CreateReplicateInvariantOpHoistingPass()); in CreateTPUBridgePipeline()
23 namespace TFDevice { namespace62 static mlir::PassRegistration<mlir::TFDevice::DecomposeResourceOps> pass(
41 namespace TFDevice { namespace181 return std::make_unique<TFDevice::ClusterOpsByPolicyPass>(); in CreateClusterOpsByPolicyPass()
36 namespace TFDevice { namespace
32 namespace TFDevice { namespace
56 namespace TFDevice { namespace
27 namespace TFDevice { namespace
33 namespace TFDevice { namespace
158 func_pm.addPass(TFDevice::CreateClusterFormationPass()); in CreateTFStandardPipeline()
79 namespace TFDevice { namespace
240 namespace TFDevice {
46 namespace TFDevice { namespace
34 namespace TFDevice { namespace
39 namespace TFDevice { namespace
250 let constructor = "TFDevice::CreateClusterConstantSinkingPass()";289 let constructor = "TFDevice::CreateClusterOutliningPass()";438 let constructor = "TFDevice::CreateMarkOpsForOutsideCompilationPass()";754 let constructor = "TFDevice::CreateClusterOpsByPolicyPass()";
1356 namespace TFDevice { namespace
117 mlir::TFDevice::CreateDecomposeResourceOpsPass()); in AddTFToTFLConversionPasses()
297 mlir::TFDevice::CreateDecomposeResourceOpsPass()); in CreateConvertMlirToXlaHloPipeline()