minimal_convolution.h

Go to the documentation of this file.

 
#ifndef OAKFIELD_MINIMAL_CONVOLUTION_H
#define OAKFIELD_MINIMAL_CONVOLUTION_H
 
#include "oakfield/operator_split.h"
 
#include <stdbool.h>
#include <stddef.h>
 
#ifdef __cplusplus
extern "C" {
#endif
 
struct SimContext;
 
enum {
    SIM_MINIMAL_CONVOLUTION_MAX_TAPS = 9,
    SIM_MINIMAL_CONVOLUTION_MAX_TAPS_2D =
        SIM_MINIMAL_CONVOLUTION_MAX_TAPS * SIM_MINIMAL_CONVOLUTION_MAX_TAPS,
};
 
typedef enum SimMinimalConvolutionMode {
    SIM_MINIMAL_CONVOLUTION_MODE_AXIS = 0,  
    SIM_MINIMAL_CONVOLUTION_MODE_SEPARABLE, 
    SIM_MINIMAL_CONVOLUTION_MODE_KERNEL_2D  
} SimMinimalConvolutionMode;
 
typedef enum SimMinimalConvolutionAxis {
    SIM_MINIMAL_CONVOLUTION_AXIS_X = 0, 
    SIM_MINIMAL_CONVOLUTION_AXIS_Y      
} SimMinimalConvolutionAxis;
 
typedef struct SimMinimalConvolutionOperatorConfig {
    size_t input_field;                              
    size_t output_field;                             
    size_t kernel_length;                            
    double kernel[SIM_MINIMAL_CONVOLUTION_MAX_TAPS]; 
    SimMinimalConvolutionMode mode;                  
    SimMinimalConvolutionAxis axis;                  
    size_t kernel_rows;                              
    size_t kernel_cols;                              
    double kernel_2d[SIM_MINIMAL_CONVOLUTION_MAX_TAPS_2D]; 
    size_t stride;                                         
    bool wrap;                    
    SimIRBoundaryPolicy boundary; 
    bool accumulate;              
    bool scale_by_dt;             
} SimMinimalConvolutionOperatorConfig;
 
SimResult sim_add_minimal_convolution_operator(struct SimContext *context,
                                               const SimMinimalConvolutionOperatorConfig *config,
                                               size_t *out_index);
 
SimResult sim_minimal_convolution_config(struct SimContext *context, size_t operator_index,
                                         SimMinimalConvolutionOperatorConfig *out_config);
 
SimResult sim_minimal_convolution_update(struct SimContext *context, size_t operator_index,
                                         const SimMinimalConvolutionOperatorConfig *config);
 
#ifdef __cplusplus
}
#endif
 
#endif /* OAKFIELD_MINIMAL_CONVOLUTION_H */