Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
H A Dtpu_rewrite_device_util.h247 bool HasModelParallelism(mlir::tf_device::ClusterOp cluster);
H A 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()
H A Dtpu_rewrite_device_util.cc485 bool HasModelParallelism(mlir::tf_device::ClusterOp cluster) { in HasModelParallelism() function