Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Ddot_handler.cc3168 const double all_gather_time_in_ms = visitor->GetCommunicationTimeInMilliSec( in PrioritizeContractingDimensionsPartitioning() local
3193 << "all_gather_time_in_ms: " << all_gather_time_in_ms in PrioritizeContractingDimensionsPartitioning()
3197 if (computation_time_in_ms == 0.0 || all_gather_time_in_ms == 0.0 || in PrioritizeContractingDimensionsPartitioning()
3203 } else if ((computation_time_in_ms <= all_gather_time_in_ms) && in PrioritizeContractingDimensionsPartitioning()
3208 return all_gather_time_in_ms > reduce_scatter_time_in_ms; in PrioritizeContractingDimensionsPartitioning()