Searched defs:reduce_function (Results 1 – 2 of 2) sorted by relevance
573 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument590 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument606 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument681 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
272 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local