Function getFNSD

src/api/cpp/common.hpp:27–38

Get the first non-zero dimension

/// Get the first non-zero dimension
static inline dim_t getFNSD(const int dim, af::dim4 dims) {
    if (dim >= 0) return dim;

    dim_t fNSD = 0;
    for (dim_t i = 0; i < 4; ++i) {
        if (dims[i] > 1) {
            fNSD = i;
            break;
        }
    }
    return fNSD;
}

namespace {
// casts from one type to another. Needed for af_half conversions specialization