ReduceOps.h

Go to the documentation of this file.

#pragma once
 
#include <limits>
#include <cstddef>
#include <utility>
 
namespace NN {
 
struct ReduceSum {
    template<typename T>
    T initial() const { return T(0); }
 
    template<typename T>
    void operator()(T& acc, const T& value) const {
        acc += value;
    }
};
 
struct ReduceMin {
    template<typename T>
    T initial() const { return std::numeric_limits<T>::max(); }
 
    template<typename T>
    void operator()(T& acc, const T& value) const {
        if (value < acc) acc = value;
    }
};
 
struct ReduceMax {
    template<typename T>
    T initial() const { return std::numeric_limits<T>::lowest(); }
 
    template<typename T>
    void operator()(T& acc, const T& value) const {
        if (value > acc) acc = value;
    }
};
 
struct ReduceArgMin {
    template<typename T>
    std::pair<int64_t, T> initial() const {
        return {0, std::numeric_limits<T>::max()};
    }
 
    template<typename T>
    void operator()(std::pair<int64_t, T>& acc, const T& value, int64_t idx) const {
        if (value < acc.second) {
            acc = {idx, value};
        }
    }
};
 
struct ReduceArgMax {
    template<typename T>
    std::pair<int64_t, T> initial() const {
        return {0, std::numeric_limits<T>::lowest()};
    }
 
    template<typename T>
    void operator()(std::pair<int64_t, T>& acc, const T& value, int64_t idx) const {
        if (value > acc.second) {
            acc = {idx, value};
        }
    }
};
}