ip.hpp

#pragma once
#include <dlprim/tensor.hpp>
#include <dlprim/context.hpp>
namespace dlprim {
namespace core {
    struct IPSettings {
        int inputs = -1;   
        int outputs = -1;  
        int optimal_batch_size = -1;  
        DataType dtype=float_data;
    };
    
    class IPForward {
    public:
        virtual ~IPForward() {}
        virtual void enqueue(Tensor &x,Tensor &w,Tensor *bias,Tensor &y,ExecutionContext const &e) = 0;
        static std::unique_ptr<IPForward> create(Context &ctx,
                                                 IPSettings const &config,
                                                 bool bias,
                                                 StandardActivations activation = StandardActivations::identity);
    };

    class IPBackwardData {
    public:
        virtual ~IPBackwardData() {}
        virtual void enqueue(Tensor &dx,Tensor &w,Tensor &dy,float factor,ExecutionContext const &e) = 0;
        static std::unique_ptr<IPBackwardData> create(Context &ctx,IPSettings const &config);
    };

    class IPBackwardFilter {
    public:
        virtual ~IPBackwardFilter() {}
        virtual void enqueue(Tensor &x,Tensor &dw,Tensor &dy,float factor,ExecutionContext const &e) = 0;
        static std::unique_ptr<IPBackwardFilter> create(Context &ctx,IPSettings const &config);
    };

} // core
} // dlprim