Searched defs:PerformReductionStep (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ | ||
| D | cpu_runtime.cc | 632 T PerformReductionStep(xla::ReductionKind reduction_kind, T a, T b) { in PerformReductionStep() function in __anonb2e2741f0511::CpuAllReduceRendezvous |
| /external/tensorflow/tensorflow/compiler/xla/service/ | ||
| D | hlo_evaluator.cc | 3603 static StatusOr<bool> PerformReductionStep( in PerformReductionStep() function |