Searched defs:reduce_function (Results 1 – 2 of 2) sorted by relevance
510 const std::function<float(float, float)>& reduce_function) { in ReduceToColArray2D() argument527 const std::function<float(float, float)>& reduce_function) { in ReduceToRowArray2D() argument543 const std::function<float(float, float)>& reduce_function) { in Reduce4DTo1D() argument618 const std::function<float(float, float)>& reduce_function) { in Reduce3DTo2D() argument
374 auto reduce_function = [&]() -> HloComputation* { in HandleBatchGroupCount() local