Function idxByndEdge

src/backend/cpu/kernel/pad_array_borders.hpp:21–37

namespace kernel {
namespace {
static inline dim_t idxByndEdge(const dim_t i, const dim_t lb, const dim_t len,
                                const af::borderType btype) {
    dim_t retVal;
    switch (btype) {
        case AF_PAD_SYM: retVal = trimIndex(i - lb, len); break;
        case AF_PAD_CLAMP_TO_EDGE:
            retVal = std::max(dim_t(0), std::min(i - lb, len - 1));
            break;
        case AF_PAD_PERIODIC: {
            dim_t rem = (i - lb) % len;
            bool cond = rem < 0;
            retVal    = cond * (rem + len) + (1 - cond) * rem;
        } break;
        default: retVal = 0; break;
    }
    return retVal;
}
}  // namespace

template<typename T>