Searched refs:PerformReductionStep (Results 1 – 2 of 2) sorted by relevance
555 out = PerformReductionStep<T>( in DoAllReduce()582 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep() function in __anonf775ca150311::CpuAllReduceRendezvous
2362 static StatusOr<bool> PerformReductionStep( in PerformReductionStep() function2461 return PerformReductionStep(is_tuple, input_index, output_index, in GenerateReduceOutputElement()