main.es6 507 字节
Newer Older
W
wangqun 已提交
1 2 3 4 5 6 7 8 9
/* eslint-disable */
/**
 * @file mul主函数
 */
export default `
// start函数
void main(void) {
    float res = 0.0;
    // 获取output的坐标
10
    ivec4 out_pos = getOutputTensorPosLIMIT_OUT();
W
wangqun 已提交
11
    for (int j = 0; j < width_shape_origin; j++) {
12 13
        float c = getValueFromTensorPosLIMIT_COUNTER_counter(out_pos[0], out_pos[1], j, out_pos[3]);
        float o = getValueFromTensorPosLIMIT_COUNTER_origin(out_pos[0], out_pos[1], out_pos[2], j);
W
wangqun 已提交
14 15 16 17 18
        res += c * o;
    }
    setOutput(res);
}
`;