bias.hpp

#pragma once
#include <dlprim/tensor.hpp>
#include <dlprim/context.hpp>
namespace dlprim {
namespace core {
    class BiasBackwardFilter {
    public:
        virtual ~BiasBackwardFilter() {}
        virtual size_t workspace() = 0;
        virtual void enqueue(Tensor &dy,Tensor &dw,Tensor &ws,float beta,ExecutionContext const &e) = 0;
        static std::unique_ptr<BiasBackwardFilter> create(Context &ctx,Shape const &dy_shape,DataType dt=float_data);
    };

    void add_bias(Tensor &t,Tensor &b,ExecutionContext const &e);
} // core
} // dlprim