Searched refs:enable_cross_replica_sharding_mirrored_variables_ (Results 1 – 2 of 2) sorted by relevance
585 static bool enable_cross_replica_sharding_mirrored_variables_; variable
2326 enable_cross_replica_sharding_mirrored_variables_); in BuildCompileNode()4029 if (enable_cross_replica_sharding_mirrored_variables_) { in GetIOTypes()4519 if (enable_cross_replica_sharding_mirrored_variables_) { in Run()4538 enable_cross_replica_sharding_mirrored_variables_ = true; member in tensorflow::DistributedTPURewritePass4551 enable_cross_replica_sharding_mirrored_variables_ = in SetDistributedTpuRewritePassOptions()