Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.h598 static bool enable_cross_replica_sharding_mirrored_variables_; variable
H A Ddistributed_tpu_rewrite_pass.cc2620 enable_cross_replica_sharding_mirrored_variables_); in BuildCompileNode()
4484 if (enable_cross_replica_sharding_mirrored_variables_) { in GetIOTypes()
5045 if (enable_cross_replica_sharding_mirrored_variables_) { in InternalRun()
5064 enable_cross_replica_sharding_mirrored_variables_ = true; member in tensorflow::DistributedTPURewritePass
5080 enable_cross_replica_sharding_mirrored_variables_ = in SetDistributedTpuRewritePassOptions()