Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h599 static bool enable_automatic_model_parallelism_; variable
H A Ddistributed_tpu_rewrite_pass.cc2246 if (node_and_sharding.has_value() && enable_automatic_model_parallelism_) { in AssignArgsAndRetvalsToCores()
2370 if (enable_automatic_model_parallelism_) { in AssignArgsAndRetvalsToCores()
5065 bool DistributedTPURewritePass::enable_automatic_model_parallelism_ = false; member in tensorflow::DistributedTPURewritePass
5082 enable_automatic_model_parallelism_ = enable_automatic_model_parallelism; in SetDistributedTpuRewritePassOptions()