| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | bridge.cc | 108 pm.addNestedPass<func::FuncOp>(TFDevice::CreateDeviceAttributeToLaunchPass()); in CreateTPUBridgePipelineImpl()
116 pm.addPass(TFDevice::CreateDecomposeResourceOpsInClusterPass()); in CreateTPUBridgePipelineImpl()
127 pm.addPass(TFDevice::CreateHostLaunchToOutsideCompiledPass()); in CreateTPUBridgePipelineImpl()
134 pm.addNestedPass<func::FuncOp>(TFDevice::CreateLaunchToDeviceAttributePass()); in CreateTPUBridgePipelineImpl()
152 pm.addPass(TFDevice::CreateResourceOpLiftingPass()); in CreateTPUBridgePipelineImpl()
162 pm.addPass(TFDevice::CreateHostLaunchToOutsideCompiledPass()); in CreateTPUBridgePipelineImpl()
166 pm.addPass(TFDevice::CreateMergeControlFlowPass()); in CreateTPUBridgePipelineImpl()
169 pm.addPass(TFDevice::CreateMarkOpsForOutsideCompilationPass()); in CreateTPUBridgePipelineImpl()
173 pm.addNestedPass<func::FuncOp>(TFDevice::CreateClusterConstantSinkingPass()); in CreateTPUBridgePipelineImpl()
175 pm.addPass(TFDevice::CreateClusterOutliningPass()); in CreateTPUBridgePipelineImpl()
[all …]
|
| D | test_cluster_ops_by_policy.cc | 24 using mlir::TFDevice::ClusteringPolicy;
25 using mlir::TFDevice::ClusteringPolicySet;
26 using mlir::TFDevice::ValueConstraint;
27 using mlir::TFDevice::ValuesConstraintSet;
|
| D | tf_device_passes.td | 109 let constructor = "TFDevice::CreateResourceOpLiftingPass()";
116 let constructor = "TFDevice::CreateResourceOpLiftingForMainFunctionPass()";
122 let constructor = "TFDevice::CreateAnnotateParameterReplicationPass()";
154 let constructor = "TFDevice::CreateDeviceAttributeToLaunchPass()";
191 let constructor = "TFDevice::CreateHostLaunchToOutsideCompiledPass()";
221 let constructor = "TFDevice::CreateDecomposeResourceOpsPass()";
233 let constructor = "TFDevice::CreateDecomposeResourceOpsInClusterPass()";
276 let constructor = "TFDevice::CreateLaunchToDeviceAttributePass()";
281 let constructor = "TFDevice::CreateXlaClusterFormationPass()";
287 let constructor = "TFDevice::CreateXlaInlineDeviceOpsPass()";
|
| D | xla_inline_device_ops.cc | 30 : public TFDevice::XlaInlineDeviceOpsPassBase<XlaInlineDeviceOpsPass> {
72 namespace TFDevice { namespace
|
| D | xla_cluster_formation.cc | 27 : public TFDevice::XlaClusterFormationPassBase<XlaClusterFormationPass> {
67 namespace TFDevice { namespace
|
| D | tf_device_passes_detail.h | 23 namespace TFDevice {
|
| D | convert_launch_func_to_tf_call.cc | 37 namespace TFDevice { namespace
|
| D | device_attribute_to_launch.cc | 27 namespace TFDevice { namespace
|
| D | annotate_parameter_replication.cc | 34 namespace TFDevice { namespace
|
| D | host_launch_to_outside_compiled.cc | 30 namespace TFDevice { namespace
|
| D | replica_id_to_device_ordinal.cc | 30 namespace TFDevice { namespace
|
| D | sink_constant.cc | 36 namespace TFDevice { namespace
|
| D | launch_to_device_attribute.cc | 29 namespace TFDevice { namespace
|
| D | mark_input_output_aliases.cc | 25 namespace TFDevice { namespace
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/ |
| D | tf_cpurt_test_passes.cc | 34 using mlir::TFDevice::ClusteringPolicySet;
35 using mlir::TFDevice::ValuesConstraintSet;
37 using mlir::TFDevice::Cluster;
38 using mlir::TFDevice::CreateClusterOp;
39 using mlir::TFDevice::FindClustersInTheBlock;
|
| D | tf_cpurt_clustering.h | 43 mlir::TFDevice::ClusteringPolicySet& policies,
51 mlir::TFDevice::ClusteringPolicySet& policies,
65 mlir::LogicalResult VerifyCluster(const mlir::TFDevice::Cluster& cluster);
|
| D | tf_cpurt_clustering.cc | 42 using mlir::TFDevice::Cluster;
43 using mlir::TFDevice::ClusteringPolicy;
44 using mlir::TFDevice::ClusteringPolicySet;
45 using mlir::TFDevice::ValueConstraint;
46 using mlir::TFDevice::ValuesConstraintSet;
|
| D | tf_cpurt_passes.cc | 666 using mlir::TFDevice::Cluster;
667 using mlir::TFDevice::ClusteringPolicySet;
668 using mlir::TFDevice::CreateClusterOp;
669 using mlir::TFDevice::FindClustersInTheBlock;
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/ |
| D | tf_jitrt_test_passes.cc | 35 using mlir::TFDevice::Cluster;
36 using mlir::TFDevice::ClusteringPolicySet;
37 using mlir::TFDevice::CreateClusterOp;
38 using mlir::TFDevice::FindClustersInTheBlock;
39 using mlir::TFDevice::ValuesConstraintSet;
|
| D | tf_jitrt_clustering.h | 58 mlir::TFDevice::ClusteringPolicySet& policies,
66 mlir::TFDevice::ClusteringPolicySet& policies,
80 mlir::LogicalResult VerifyCluster(const mlir::TFDevice::Cluster& cluster);
|
| D | tf_jitrt_clustering_pass.cc | 39 using mlir::TFDevice::Cluster;
40 using mlir::TFDevice::ClusteringPolicySet;
41 using mlir::TFDevice::CreateClusterOp;
42 using mlir::TFDevice::FindClustersInTheBlock;
|
| D | tf_jitrt_clustering.cc | 43 using mlir::TFDevice::Cluster;
44 using mlir::TFDevice::ClusteringPolicy;
45 using mlir::TFDevice::ClusteringPolicySet;
46 using mlir::TFDevice::ValueConstraint;
47 using mlir::TFDevice::ValuesConstraintSet;
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | dtensor_mlir_passes.cc | 237 pm->addPass(mlir::TFDevice::CreateClusterOutliningPass()); in CreateDTensorMLIRPass()
255 pm->addPass(mlir::TFDevice::CreateMergeControlFlowPass()); in CreateDTensorMLIRPass()
265 mlir::TFDevice::CreateDecomposeResourceOpsPass()); in CreateDTensorMLIRPass()
269 mlir::TFDevice::CreateClusterConstantSinkingPass()); in CreateDTensorMLIRPass()
275 pm->addPass(mlir::TFDevice::CreateResourceOpLiftingPass()); in CreateDTensorMLIRPass()
276 pm->addPass(mlir::TFDevice::CreateClusterOutliningPass()); in CreateDTensorMLIRPass()
289 pm->addPass(mlir::TFDevice::CreateMarkInputOutputAliasesPass()); in CreateDTensorMLIRPass()
328 mlir::TFDevice::CreateLaunchToDeviceAttributePass()); in CreateDTensorMLIRPass()
|
| /external/tensorflow/tensorflow/compiler/mlir/ |
| D | tf_mlir_reduce_main.cc | 57 mlir::TFDevice::registerTensorFlowDevicePasses(); in main()
|
| D | tf_mlir_opt_main.cc | 46 mlir::TFDevice::registerTensorFlowDevicePasses(); in main()
|