Searched refs:enable_automatic_model_parallelism_ (Results 1 – 2 of 2) sorted by relevance
586 static bool enable_automatic_model_parallelism_; variable
2032 if (node_and_sharding.has_value() && enable_automatic_model_parallelism_) { in AssignArgsAndRetvalsToCores()2146 if (enable_automatic_model_parallelism_) { in AssignArgsAndRetvalsToCores()4539 bool DistributedTPURewritePass::enable_automatic_model_parallelism_ = false; member in tensorflow::DistributedTPURewritePass4553 enable_automatic_model_parallelism_ = enable_automatic_model_parallelism; in SetDistributedTpuRewritePassOptions()