main.es6

/* eslint-disable */
/**
 * @file 主函数
 * @author chenhaoze
 */
export default `
    // start函数
    void main(void) {
        ivec4 oPos = getOutputTensorPosLIMIT_OUT();
        int x = oPos.a;
        int c = oPos.g;
        int y = oPos.b;
        int b = oPos.r;
        float res = 0.0;
        int temp_x = 0;
        int temp_y = 0;
        float o = 0.0;
        float f = 0.0;
        
        // 获取output的坐标
        int oTensorChannel = int(c * groups / channel_out) * channel_origin;
        int oy = y - padTop;
        for (int fy = 0; fy < height_shape_filter; fy++) {
            if (oy < 0) {
                oy += dilation_v;
                continue;
            }
            int ox = x - padLeft;
            for (int fx = 0; fx < width_shape_filter; fx++) {

                if (ox < 0) {
                    ox += dilation_h;
                    continue;
                }
                // channel计算
                for (int j = 0; j < channel_origin; j++) {
                	if (int(mod(float(ox), float(stride_h))) == 0 && int(mod(float(oy), float(stride_v))) == 0) {
						temp_x = int(floor(float(ox) / float(stride_h)));
						temp_y = int(floor(float(oy) / float(stride_v)));
                        if (temp_x < width_shape_origin && temp_y < height_shape_origin){
						    o = getValueFromTensorPosLIMIT_ORIGIN_origin(b, j, temp_y, temp_x);
                            f = getValueFromTensorPosLIMIT_FILTER_filter(j, c, height_shape_filter-1-fy, width_shape_filter-1-fx);
                            res += f * o;
                        }
					}
                }
                ox += dilation_h;
            }
            oy += dilation_v;
        }
        setOutput(float(res));
    }
`;