conf.es6 1.1 KB
Newer Older
Y
yangmingming 已提交
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 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50
/* eslint-disable */
/**
 * @file pool2d的配置文件
 * @author yangmingming zhangmiao06
 */
export default {
    dep: [
        {
            func: 'getValueFromTensorPosPacked',
            conf: {
                TENSOR_NAME: 'origin'
            }
        }
    ],
    conf: [
        'KSIZE_X',
        'KSIZE_Y',
        'TYPE_POOL',

        'WIDTH_SHAPE_ORIGIN',
        'HEIGHT_SHAPE_ORIGIN',
        'LENGTH_SHAPE_ORIGIN',
        'WIDTH_TEXTURE_ORIGIN',
        'HEIGHT_TEXTURE_ORIGIN',
        'CHANNEL_ORIGIN',
        'OFFSET_X_ORIGIN',
        'OFFSET_Y_ORIGIN',

        'WIDTH_SHAPE_OUT',
        'HEIGHT_SHAPE_OUT',
        'WIDTH_TEXTURE_OUT',
        'HEIGHT_TEXTURE_OUT',
        'CHANNEL_OUT',
        'OFFSET_Y_OUT',

        'STRIDES_X',
        'STRIDES_Y',
        'PADDING_X',
        'PADDING_Y'
    ],
    input: [
        // texture类型,若添加from: 'prev', 表示读取上一个op的产出
        {
            tensor: 'origin',
            variable: 'texture',
            setter: 'initTexture',
            type: 'texture'
        }
    ]
};