Searched defs:reduce_function (Results 1 – 2 of 2) sorted by relevance
508 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument525 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument541 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument616 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
369 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local