params.es6 1.4 KB
Newer Older
W
wangqun 已提交
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30
/* eslint-disable */
/**
 * @file 参数文件
 * @author chenhaoze
 */
export default `
    // conv2d的input数据
    
    // 常量
    // 卷积核
    const int length_shape_filter = LENGTH_SHAPE_FILTER;
    const int width_shape_filter = WIDTH_SHAPE_FILTER;
    const int height_shape_filter = HEIGHT_SHAPE_FILTER;
    const int width_texture_filter = WIDTH_TEXTURE_FILTER;
    const int height_texture_filter = HEIGHT_TEXTURE_FILTER;
    const int channel_filter = CHANNEL_FILTER;
    
    // 输入数据
    const int width_shape_origin = WIDTH_SHAPE_ORIGIN;
    const int height_shape_origin = HEIGHT_SHAPE_ORIGIN;
    const int length_shape_origin = LENGTH_SHAPE_ORIGIN;
    const int width_texture_origin = WIDTH_TEXTURE_ORIGIN;
    const int height_texture_origin = HEIGHT_TEXTURE_ORIGIN;
    const int channel_origin = CHANNEL_ORIGIN;
    
    // 计算相关
    // 拆分步长
    const int stride_h = int(STRIDES_X);
    const int stride_v = int(STRIDES_Y);
    // padding的数目
31 32 33
    const int padLeft = WIDTH_SHAPE_FILTER - PADDINGS_X - 1;
    const int padTop = HEIGHT_SHAPE_FILTER - PADDINGS_Y - 1;

W
wangqun 已提交
34
    // dilation膨胀系数
W
wangqun 已提交
35 36
    const int dilation_h = DILATIONS_X;
    const int dilation_v = DILATIONS_Y;
W
wangqun 已提交
37 38 39 40 41 42 43 44 45 46
    // groups
    const int groups = GROUPS;
    
    // uniform变量
    // 卷积核
    uniform sampler2D texture_filter;
    
    // 输入数据
    uniform sampler2D texture_origin;
`;