Home
last modified time | relevance | path

Searched defs:PerformReductionStep (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_runtime.cc632 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep() function in __anon8f39799f0511::CpuAllReduceRendezvous
653 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep() function in __anon8f39799f0511::CpuAllReduceRendezvous
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_evaluator.cc3603 static StatusOr<bool> PerformReductionStep( in PerformReductionStep() function