Home
last modified time | relevance | path

Searched refs:kNumCoresPerReplicaAttr (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util_test.cc653 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
674 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
694 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
714 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
741 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
796 cluster->setAttr(kNumCoresPerReplicaAttr, in TEST()
Dtpu_rewrite_device_util.h38 extern const char* const kNumCoresPerReplicaAttr;
Dtpu_rewrite_device_util.cc45 const char* const kNumCoresPerReplicaAttr = "num_cores_per_replica"; variable
499 tensorflow::kNumCoresPerReplicaAttr); in GetHostDeviceOutsideComputation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_cluster_formation.cc55 constexpr char kNumCoresPerReplicaAttr[] = "num_cores_per_replica"; variable
571 .get(kNumCoresPerReplicaAttr) in FormClustersInBlock()
Dtpu_rewrite_pass.cc577 tensorflow::kNumCoresPerReplicaAttr); in Rewrite()
580 CreateMissingAttributeMsg(tensorflow::kNumCoresPerReplicaAttr)); in Rewrite()