Home
last modified time | relevance | path

Searched refs:kNumReplicasAttr (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc56 constexpr char kNumReplicasAttr[] = "num_replicas"; variable
348 return cluster.emitError() << "requires '" << kNumReplicasAttr in ReplicateCluster()
563 auto num_replicas = cluster_metadata->getSecond().get(kNumReplicasAttr); in FormClustersInBlock()
566 << "requires '" << kNumReplicasAttr << "' int attribute"; in FormClustersInBlock()
585 cluster.removeAttr(kNumReplicasAttr); in FormClustersInBlock()
Dtpu_rewrite_pass.cc67 constexpr char kNumReplicasAttr[] = "num_replicas"; variable