Home
last modified time | relevance | path

Searched refs:CreateClusterOp (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/
Dtf_cpurt_test_passes.cc38 using mlir::TFDevice::CreateClusterOp;
62 CreateClusterOp(cluster, {}); in runOnFunction()
Dtf_cpurt_passes.cc668 using mlir::TFDevice::CreateClusterOp;
723 CreateClusterOp(cluster, policy); in runOnFunction()
/external/tensorflow/tensorflow/compiler/mlir/tfrt/jit/transforms/
Dtf_jitrt_test_passes.cc37 using mlir::TFDevice::CreateClusterOp;
56 CreateClusterOp(cluster, {}); in runOnOperation()
Dtf_jitrt_clustering_pass.cc41 using mlir::TFDevice::CreateClusterOp;
151 CreateClusterOp(cluster, policy); in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dcluster_ops_by_policy.h234 tf_device::ClusterOp CreateClusterOp(Cluster& cluster, StringAttr policy = {});
Dtpu_cluster_formation.cc319 tf_device::ClusterOp CreateClusterOp( in CreateClusterOp() function
615 tf_device::ClusterOp cluster = CreateClusterOp( in FormClustersInBlock()
Dcluster_ops_by_policy.cc539 tf_device::ClusterOp CreateClusterOp(Cluster &cluster, StringAttr policy) { in CreateClusterOp() function