Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.h62 bool cache_all_gather = true; member
Dspmd_partitioner.cc352 if (!is_to_replicate || state_.partitioner->options().cache_all_gather) { in Reshard()
362 if (!is_to_replicate || state_.partitioner->options().cache_all_gather) { in Reshard()
825 if (state_.partitioner->options().cache_all_gather) { in Replicate()
847 if (state_.partitioner->options().cache_all_gather) { in Replicate()