bias_add.hpp

Go to the documentation of this file.

//=======================================================================
// Copyright (c) 2014-2023 Baptiste Wicht
// Distributed under the terms of the MIT License.
// (See accompanying file LICENSE or copy at
//  http://opensource.org/licenses/MIT)
//=======================================================================

#pragma once

namespace etl::impl::standard {

template <typename A, typename B, typename C>
void bias_add_4d(const A& lhs, const B& rhs, C&& c) {
    for (size_t i = 0; i < etl::dim<0>(lhs); ++i) {
        for (size_t j = 0; j < etl::dim<1>(lhs); ++j) {
            for (size_t k = 0; k < etl::dim<2>(lhs); ++k) {
                for (size_t l = 0; l < etl::dim<3>(lhs); ++l) {
                    c(i, j, k, l) = lhs(i, j, k, l) + rhs(j);
                }
            }
        }
    }
}

template <typename A, typename B, typename C>
void bias_add_2d(const A& lhs, const B& rhs, C&& c) {
    for (size_t i = 0; i < etl::dim<0>(lhs); ++i) {
        for (size_t j = 0; j < etl::dim<1>(lhs); ++j) {
            c(i, j) = lhs(i, j) + rhs(j);
        }
    }
}

} //end of namespace etl::impl::standard