Searched refs:PerformReductionStep (Results 1 – 2 of 2) sorted by relevance
525 out = PerformReductionStep<T>( in DoAllReduce()552 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep() function in __anon95669c100311::CpuAllReduceRendezvous
2298 static StatusOr<bool> PerformReductionStep( in PerformReductionStep() function2397 return PerformReductionStep(is_tuple, input_index, output_index, in GenerateReduceOutputElement()