Home
last modified time | relevance | path

Searched refs:HasModelParallelism (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtpu_rewrite_device_util.h247 bool HasModelParallelism(mlir::tf_device::ClusterOp cluster);
Dtpu_rewrite_device_util_test.cc643 EXPECT_FALSE(HasModelParallelism(cluster)); in TEST()
661 EXPECT_TRUE(HasModelParallelism(cluster)); in TEST()
680 EXPECT_FALSE(HasModelParallelism(cluster)); in TEST()
Dtpu_rewrite_device_util.cc485 bool HasModelParallelism(mlir::tf_device::ClusterOp cluster) { in HasModelParallelism() function