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.h135 bool enable_cross_replica_sharding_mirrored_variables,
H A Ddistributed_tpu_rewrite_pass.cc5073 bool enable_cross_replica_sharding_mirrored_variables, in SetDistributedTpuRewritePassOptions() argument
5081 enable_cross_replica_sharding_mirrored_variables; in SetDistributedTpuRewritePassOptions()