Home
last modified time | relevance | path

Searched refs:TFDevice (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dbridge.cc50 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()
Ddecompose_resource_ops_pass.cc23 namespace TFDevice { namespace
62 static mlir::PassRegistration<mlir::TFDevice::DecomposeResourceOps> pass(
Dcluster_ops_by_policy_pass.cc41 namespace TFDevice { namespace
181 return std::make_unique<TFDevice::ClusterOpsByPolicyPass>(); in CreateClusterOpsByPolicyPass()
Dsink_constant.cc36 namespace TFDevice { namespace
Dannotate_parameter_replication.cc32 namespace TFDevice { namespace
Dlaunch_to_device_attribute.cc56 namespace TFDevice { namespace
Dtf_device_replication_pass.cc27 namespace TFDevice { namespace
Dcluster_outlining.cc32 namespace TFDevice { namespace
Dreplicate_invariant_op_hoisting.cc33 namespace TFDevice { namespace
Doptimize.cc158 func_pm.addPass(TFDevice::CreateClusterFormationPass()); in CreateTFStandardPipeline()
Dparallel_execute_to_islands.cc79 namespace TFDevice { namespace
Dcluster_formation.cc36 namespace TFDevice { namespace
Dpasses.h240 namespace TFDevice {
Dreplicate_to_island.cc46 namespace TFDevice { namespace
Dmerge_control_flow.cc34 namespace TFDevice { namespace
Dmark_ops_for_outside_compilation.cc39 namespace TFDevice { namespace
Dtf_passes.td250 let constructor = "TFDevice::CreateClusterConstantSinkingPass()";
289 let constructor = "TFDevice::CreateClusterOutliningPass()";
438 let constructor = "TFDevice::CreateMarkOpsForOutsideCompilationPass()";
754 let constructor = "TFDevice::CreateClusterOpsByPolicyPass()";
Dresource_op_lifting.cc1356 namespace TFDevice { namespace
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dtf_tfl_passes.cc117 mlir::TFDevice::CreateDecomposeResourceOpsPass()); in AddTFToTFLConversionPasses()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc297 mlir::TFDevice::CreateDecomposeResourceOpsPass()); in CreateConvertMlirToXlaHloPipeline()